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

Cách viết pythonic của mã sau đây là gì?

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

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

Tôi có một bộ nhớ mơ hồ rằng việc khai báo rõ ràng về forvòng lặp có thể tránh được và được viết trong ifđiều kiện. Điều này có đúng không?

Mặc dù không được 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.

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

Chỉ dùng:

if file_name.endswith(tuple(extensions)):

Lấy một phần mở rộng từ tệp và xem nếu nó nằm trong bộ tiện ích 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ì độ phức tạp thời gian để tra cứu trong các bộ là O (1) ( tài liệu ).

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

Các phương thức 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 của 7 lần chạy, mỗi vòng 1000000)

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, mỗi vòng 1000000)

Tôi có cái này:

def has_extension(filename, extension):

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

Tôi vừa đi qua cái này, trong khi tìm kiếm thứ khác.

Tôi sẽ khuyên bạn nên đi với các phương pháp trong osgói. Điều này là do bạn có thể làm cho nó chung chung hơn, bù đắp cho bất kỳ trường hợp kỳ lạ nào.

Bạn có thể làm một cái gì đó như:

import os

the_file = 'aaaa/bbbb/ccc.ddd'

extensions_list = ['ddd', 'eee', 'fff']

if os.path.splitext(the_file)[-1] in extensions_list:
    # Do your thing.

Một khả năng khác có thể là sử dụng câu lệnh IN:

extensions = ['.mp3','.avi']
file_name  = 'test.mp3'
if "." in file_name and file_name[file_name.rindex("."):] in extensions:
    print(True)

một cách khác có thể trả về danh sách các chuỗi phù hợp là

sample = "alexis has the control"
matched_strings = filter(sample.endswith, ["trol", "ol", "troll"])
print matched_strings
['trol', 'ol']