如何在列表理解中使用重新匹配对象

Question

我有一个函数可以从字符串列表中挑选出块并将它们作为另一个列表返回：

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

有没有办法将其重新表述为列表理解？显然它是相当清楚的；只是好奇。

---

感谢那些做出贡献的人，特别提到@Alex。这是我最终得到的浓缩版；正则表达式匹配方法作为“预提升”参数传递给 filterPick：

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]

“技巧”是for m in [regex.search(l)] 部分——这就是你如何在列表理解中“分配”一个需要多次使用的值——只添加这样一个子句，对象“迭代”在包含您要“分配”给它的一个值的单项列表上。有些人认为这在风格上是可疑的，但我有时觉得它很实用。

Answer 2

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

Answer 3

可以缩短一点

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

您可以将所有内容放在一行中，但这意味着您必须将每一行匹配两次，这会降低效率。

Answer 4

从Python 3.8 开始，并引入assignment expressions (PEP 572)（:= 运算符），可以在列表解析中使用局部变量以避免多次调用同一个表达式：

# items = ["foo", "bar", "baz", "qurx", "bother"]
[(x, match.group(1)) for x in items if (match := re.compile('(a|r$)').search(x))]
# [('bar', 'a'), ('baz', 'a'), ('bother', 'r')]

这个：

• 将re.compile('(a|r$)').search(x) 的评估命名为变量match（即None 或Match 对象）
• 使用此 match 命名表达式（None 或 Match）过滤掉不匹配的元素
• 并通过提取第一组 (match.group(1)) 在映射值中重复使用 match。

Answer 5

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

这也适用于您在列表中搜索的搜索查询

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

'b' in P` 返回真