检查字符串是否以列表中的某个字符串结尾

问题描述：

下列代码的 Python 编写方式是什么？

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

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

我依稀记得for可以避免显式声明循环并将其写在if条件中。这是真的吗？

解决方案 1：

虽然不太为人所知，但str.endswith也接受元组。您无需循环。

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

解决方案 2：

只需使用：

if file_name.endswith(tuple(extensions)):

解决方案 3：

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

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

解决方案 4：

有两种方法：正则表达式和字符串（str）方法。

字符串方法通常更快（〜2倍）。

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 次循环）

解决方案 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.

解决方案 6：

我有这个：

def has_extension(filename, extension):

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

解决方案 7：

另一种可能性是利用 IN 语句：

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