查找前x匹配re.findall

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

要找到N項匹配和停止，你可以使用re.finditer和itertools.islice：

>>> import itertools as IT 
>>> [item.group() for item in IT.islice(re.finditer(r'\d', text), 3)] 
['1', '2', '3'] 

Answer 2

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