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

编写下面代码的python方式是什么?

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


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

我有一个模糊的记忆，for循环的显式声明可以避免，并写入if条件。这是真的吗?

184187 次浏览
小开

只使用:

if file_name.endswith(tuple(extensions)):
小开
最佳答案

虽然并不广为人知，但str.endswith也接受元组。你不需要循环。

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

我有这个:

def has_extension(filename, extension):


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

有两种方法:正则表达式和字符串(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次)

小开

我在找别的东西的时候，偶然发现了这个。

我建议使用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.
小开

另一种可能是使用IN语句:

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

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

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

提交答案