Có cách nào tốt hơn để viết các câu lệnh "hoặc" liên tiếp trong Python không?

Question

câu hỏi đơn giản mà tôi không thể tìm thấy bất kỳ câu trả lời "tốt đẹp" của bản thân mình:

Hãy nói rằng tôi có điều kiện sau đây:

if 'foo' in mystring or 'bar' in mystring or 'hello' in mystring: 
    # Do something 
    pass 

Trường hợp số or tuyên bố có thể khá dài tùy thuộc vào tình huống.

Có cách nào "đẹp hơn" (hơn Pythonic) bằng cách viết này, mà không bị mất hiệu suất?

Nếu nghĩ đến việc sử dụng any() nhưng cần danh sách các phần tử giống như boolean, vì vậy tôi sẽ phải xây dựng danh sách đó trước tiên (đánh giá ngắn mạch trong quá trình), vì vậy tôi đoán nó kém hiệu quả hơn.

Cảm ơn bạn rất nhiều.

Answer 1

Một cách có thể là

if any(s in mystring for s in ('foo', 'bar', 'hello')): 
    pass 

Điều bạn lặp qua là một tuple, được xây dựng dựa trên tổng hợp các chức năng, vì vậy nó không phải là thua kém phiên bản gốc của bạn.

Nếu bạn lo ngại rằng các tuple sẽ trở nên quá dài, bạn có thể làm

def mystringlist(): 
    yield 'foo' 
    yield 'bar' 
    yield 'hello' 
if any(s in mystring for s in mystringlist()): 
    pass 

Answer 2

này nghe có vẻ như một công việc cho một regex.

import re 

if re.search("(foo|bar|hello)", mystring): 
    # Do something 
    pass 

Nó cũng sẽ nhanh hơn. Đặc biệt là nếu bạn biên dịch regex trước thời hạn.

Nếu bạn đang tạo biểu thức chính quy tự động, bạn có thể sử dụng re.escape() để đảm bảo không có ký tự đặc biệt nào phá vỡ regex của bạn. Ví dụ, nếu words là một danh sách các chuỗi bạn muốn tìm kiếm, bạn có thể tạo ra mô hình của bạn như thế này:

pattern = "(%s)" % ("|".join(re.escape(word) for word in words),) 

Bạn cũng nên lưu ý rằng nếu bạn có m từ và chuỗi của bạn có n ký tự, ban đầu của bạn mã có độ phức tạp O(n*m), trong khi cụm từ thông dụng có độ phức tạp O(n). Mặc dù Python regex không thực sự là lý thuyết comp-sci thường xuyên biểu thức, và không phải là luôn luônO(n) phức tạp, trong trường hợp này đơn giản họ đang có.

Answer 3

Vì bạn đang xử lý từng từ với mystring, chắc chắn bạn có thể sử dụng tính năng bí ẩn làm bộ. Sau đó chỉ cần lấy ngã tư giữa các thiết lập có chứa các từ trong mystring và các nhóm đối tượng của chữ:

In [370]: mystring=set(['foobar','barfoo','foo']) 

In [371]: mystring.intersection(set(['foo', 'bar', 'hello'])) 
Out[371]: set(['foo']) 

của bạn logic 'hoặc' là các thành viên của giao điểm của hai bộ.

Sử dụng bộ cũng nhanh hơn. Dưới đây là thời gian tương đối vs một máy phát điện và biểu thức chính quy:

f1: generator to test against large string 
f2: re to test against large string 
f3: set intersection of two sets of words 

    rate/sec  f2  f1  f3 
f2 101,333  -- -95.0% -95.5% 
f1 2,026,329 1899.7%  -- -10.1% 
f3 2,253,539 2123.9% 11.2%  -- 

Vì vậy, một máy phát điện và các in hoạt động là 19x nhanh hơn so với một biểu thức chính quy và một ngã tư bộ là 21x nhanh hơn so với một regex và nhanh hơn so với một máy phát điện 11%.

Đây là mã đã tạo ra thời gian:

import re 

with open('/usr/share/dict/words','r') as fin: 
    set_words={word.strip() for word in fin} 

s_words=' '.join(set_words) 
target=set(['bar','foo','hello']) 
target_re = re.compile("(%s)" % ("|".join(re.escape(word) for word in target),)) 

gen_target=(word for word in ('bar','foo','hello')) 

def f1(): 
    """ generator to test against large string """   
    if any(s in s_words for s in gen_target): 
     return True 

def f2(): 
    """ re to test against large string """ 
    if re.search(target_re, s_words): 
     return True 

def f3(): 
    """ set intersection of two sets of words """ 
    if target.intersection(set_words): 
     return True 

funcs=[f1,f2,f3] 
legend(funcs) 
cmpthese(funcs)   

Answer 4

Nếu bạn có một danh tiếng của sản phẩm để kiểm tra chống lại, bạn cũng có thể viết nó như

if mystring in ['foo', 'bar', 'hello']: 

Bạn không thể có được lợi ích của việc đảm bảo thứ tự so sánh (Tôi không nghĩ rằng Python là cần thiết để kiểm tra các yếu tố danh sách từ trái sang phải) nhưng đó chỉ là vấn đề nếu bạn biết 'foo' có nhiều khả năng hơn 'bar'.