python - 从字符串列表中获取子字符串与某个正则表达式匹配的子字符串列表

Question

此问题适用于 Python 3.6+(但请随意为其他读者回答较低版本的 Python)。

我想从与正则表达式匹配的每个字符串中提取一个子字符串。

假设我有以下内容:

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

我想要匹配 v-02-\d\d\d 的所有字符串的最后 3 位数字，即:

['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']