檢查字符串是否以列表中的字符串之一結尾

Question

編寫以下代碼的pythonic方式是什么？

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

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

我有一個模糊的記憶，可以避免for循環的顯式聲明並將其寫入if條件中。 這是真的？

Answer 1

雖然並不廣為人知，但str.endswith也接受一個元組。 你不需要循環。

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

Answer 2

只需使用：

if file_name.endswith(tuple(extensions)):

Answer 3

從文件中獲取擴展名，看看它是否在擴展集中：

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

使用集合因為集合中查找的時間復雜度為O（1）（ docs ）。

Answer 4

有兩種方法：正則表達式和字符串（str）方法。

字符串方法通常更快（~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（平均值±標准偏差，7次運行，每次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（平均值±標准偏差，7次運行，每次1000000次循環）

Answer 5

我只是想到了這個，同時尋找其他東西。

我建議使用os包中的方法。 這是因為你可以使它更通用，補償任何奇怪的情況。

你可以這樣做：

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.

Answer 6

我有這個：

def has_extension(filename, extension):

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

Answer 7

另一種可能性是使用IN語句：

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

Answer 8

另一種可以返回匹配字符串列表的方法是

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