Comment utiliser des objets re match dans une compréhension de la liste

Question

J'ai une fonction pour choisir des morceaux à partir d'une liste de chaînes et de les retourner comme une autre liste:

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

est-il un moyen de reformuler cela comme une liste compréhension? De toute évidence, c'est assez clair tel quel; juste curieux.

---

Merci à ceux qui ont contribué, une mention spéciale pour @Alex. Voici une version condensée de ce que j'ai fini avec; la méthode de match regex est passé à filterPick comme paramètre « pré-hissés »:

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] 

Le « truc » est la for m in [regex.search(l)] partie - c'est comment vous « assigner » une valeur que vous devez utiliser plus d'une fois, dans une liste de compréhension - ajouter juste une telle clause, où l'objet "itère" sur une liste d'un seul élément contenant la valeur que vous voulez lui "assigner". Certains trouvent cela stylistiquement douteux, mais je le trouve parfois pratique.

Answer 2

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

Answer 3

Il pourrait être un peu raccourcies

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

Vous pouvez le mettre en une seule ligne, mais cela voudrait dire que vous devez faire correspondre chaque ligne deux fois ce qui serait un peu moins efficace.

Answer 4

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

Cela fonctionne aussi avec une requête de recherche vous pourchassant dans une liste

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

'b' dans p` retourne vrai