Tìm các kết quả phù hợp đầu tiên với re.findall

Question

Tôi cần giới hạn re.findall để tìm 3 kết quả phù hợp đầu tiên và sau đó dừng lại.

ví dụ

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

sau đó tôi nhận được:

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

và tôi muốn:

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

Answer 1

Để tìm N phù hợp và dừng, bạn có thể sử dụng re.finditer và 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 trả về một danh sách, vì vậy giải pháp đơn giản nhất là nên chỉ sử dụng slicing :

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