从字符串列表中获取子字符串列表，其中子字符串与某个正则表达式匹配

Question

这个问题是针对 Python 3.6+ 的（但也可以为其他读者回答较低的 Python）。我想从匹配正则表达式的每个字符串中提取一个子字符串。说我有以下几点：a = ['v-01-001', 'v-01-002', 'v-02-001', 'v-02-002', 'v-02-003', 'v-03-001']我想要所有匹配的字符串的最后 3 位数字v-02-\d\d\d，即：['001', '002', '003']我天真的尝试：[x[1] for x in list(map(lambda i: re.search(r'v-02-(\d\d\d)', i), a)) if x]你能想出更优雅的东西吗？

Answer 1

你可以这样做：

import re

a = ['v-01-001', 'v-01-002', 'v-02-001', 'v-02-002', 'v-02-003', 'v-03-001']

pattern = re.compile('v-02-(\d{3})$')

print([m.group(1) for m in map(pattern.match, a) if m])

输出

['001', '002', '003']

你也可以使用finditer：

print([m.group(1) for ms in map(pattern.finditer, a) for m in ms])

输出

['001', '002', '003']

Answer 2

四种方法可以做到这一点。

第一个只是一个常规的 'ole 循环：

li=[]

for s in a:

    m = re.search(r'v-02-(\d\d\d)', s)

    if m:

        li.append(m.group(1))

 # li=['001', '002', '003']

在列表理解中对同一正则表达式的两次调用中的第二次：

>>> [re.search(r'v-02-(\d\d\d)', s).group(1) for s in a if re.search(r'v-02-(\d\d\d)', s)]

['001', '002', '003']

三是使用map：

>>> [m.group(1) for m in map(lambda s: re.search(r'v-02-(\d\d\d)', s), a) if m]

['001', '002', '003']

最后，您可以将列表展平.join，然后使用findall：

>>> re.findall(r'\bv-02-(\d\d\d)\b', '\t'.join(a))

['001', '002', '003']

或者，使用\nand re.Mvs two \b：

>>> re.findall(r'^v-02-(\d\d\d)$', '\n'.join(a), flags=re.M)

['001', '002', '003']

如果我正在编写这段代码，我可能会以相同的顺序编写它。

我想，在旁观者的眼里，什么被认为是更优雅的。我认为最后一个更优雅。

您还可以跳过正则表达式并使用 Python 的字符串方法：

>>> prefix='v-02-'

>>> [e[len(prefix):] for e in filter(lambda s: s.startswith(prefix),a)]

['001', '002', '003']

如果在这种情况下很重要，那可能是最快的。

2019 年 12 月，会有更优雅的选择。根据PEP 572 中的定义，您将能够使用赋值语句，以便您可以在一个步骤中分配匹配项并测试匹配项：

[m.group(1) for s in a if (m:=re.search(r'v-02-(\d\d\d)', s))]