Kiểm tra xem chuỗi có kết thúc bằng một trong các chuỗi từ danh sách

Question

Cách viết mã pythonic sau đây không?

extensions = ['.mp3','.avi'] 
file_name = 'test.mp3' 

for extension in extensions: 
    if file_name.endswith(extension): 
     #do stuff 

Tôi có một ký ức mơ hồ rằng việc kê khai rõ ràng của for vòng lặp có thể tránh được và được ghi trong điều kiện if. Điều này có đúng không?

Answer 1

Mặc dù không biết đến rộng rãi, str.endswith cũng chấp nhận một tuple. Bạn không cần phải lặp lại.

>>> 'test.mp3'.endswith(('.mp3', '.avi')) 
True 

Answer 2

Chỉ cần sử dụng:

if file_name.endswith(tuple(extensions)): 

Answer 3

Đi một phần mở rộng từ các tập tin và xem nếu nó nằm trong tập các phần mở rộng:

>>> import os 
>>> extensions = set(['.mp3','.avi']) 
>>> file_name = 'test.mp3' 
>>> extension = os.path.splitext(file_name)[1] 
>>> extension in extensions 
True 

Sử dụng một bộ vì thời gian phức tạp để tra cứu trong bộ là O (1) (docs).

Answer 4

Tôi có điều này:

def has_extension(filename, extension): 

    ext = "." + extension 
    if filename.endswith(ext): 
     return True 
    else: 
     return False 

Answer 5

Có hai cách: biểu thức thông thường và chuỗi (str) phương pháp.

Phương pháp chuỗi thường nhanh hơn (~ 2x).

import re, timeit 
p = re.compile('.*(.mp3|.avi)$', re.IGNORECASE) 
file_name = 'test.mp3' 
print(bool(t.match(file_name)) 
%timeit bool(t.match(file_name) 

792 ns ± 1,83 ns mỗi vòng lặp (trung bình ± std. Dev. 7 chạy, 1000000 vòng mỗi)

file_name = 'test.mp3' 
extensions = ('.mp3','.avi') 
print(file_name.lower().endswith(extensions)) 
%timeit file_name.lower().endswith(extensions) 

274 ​​ns ± 4,22 ns mỗi vòng lặp (trung bình ± std dev.. của 7 lần chạy, 1000000 vòng lặp)