Cómo utilizar objetos de los partidos re en una lista por comprensión

Question

Tengo una función para escoger trozos de una lista de cadenas y devolverlos como otra lista:

def filterPick(lines,regex): 
    result = [] 
    for l in lines: 
     match = re.search(regex,l) 
     if match: 
      result += [match.group(1)] 
    return result 

¿Hay una manera de reformular esto como una lista ¿comprensión? Obviamente está bastante claro como está; Sólo curioso.

---

Gracias a los que contribuyeron, mención especial para @Alex. Aquí hay una versión condensada de lo que terminé; el método de coincidencia de expresiones regulares se pasa a filterPick como un parámetro de "pre-izado":

import re 

def filterPick(list,filter): 
    return [ (l, m.group(1)) for l in list for m in (filter(l),) if m] 

theList = ["foo", "bar", "baz", "qurx", "bother"] 
searchRegex = re.compile('(a|r$)').search 
x = filterPick(theList,searchRegex) 

>> [('bar', 'a'), ('baz', 'a'), ('bother', 'r')] 

Answer 1

[m.group(1) for l in lines for m in [regex.search(l)] if m] 

El "truco" es la parte for m in [regex.search(l)] - así es como se "asigna" un valor que es necesario utilizar más de una vez, dentro de una lista de comprensión, agregue solo una cláusula, donde el objeto "itera" sobre una lista de un solo elemento que contiene el único valor que quiere "asignarle". Algunos lo consideran estilísticamente dudoso, pero a veces me resulta práctico.

Answer 2

return [m.group(1) for m in (re.search(regex, l) for l in lines) if m] 

Answer 3

Podría ser acortado un poco

def filterPick(lines, regex): 
    matches = map(re.compile(regex).match, lines) 
    return [m.group(1) for m in matches if m] 

se puede poner todo en una sola línea, pero eso significaría que tendría que coincidir con cada línea doble de lo que sería un poco menos eficiente.

Answer 4

>>> "a" in "a visit to the dentist" 
True 
>>> "a" not in "a visit to the dentist" 
False 

Eso también funciona con una consulta de búsqueda que estés cazando en una lista

`P = 'a', 'b', 'c'

'b' en P` devuelve verdadero