python - 使用 re.findall 查找前 x 个匹配项

Question

我需要限制 re.findall 找到前 3 个匹配项然后停止。

例如

text = 'some1 text2 bla3 regex4 python5'
re.findall(r'\d',text)

然后我得到:

['1', '2', '3', '4', '5']

我想要:

['1', '2', '3']

Answer 1

re.findall 返回一个列表，所以最简单的解决方案就是使用 slicing :

>>> import re
>>> text = 'some1 text2 bla3 regex4 python5'
>>> re.findall(r'\d', text)[:3]  # Get the first 3 items
['1', '2', '3']
>>>