检查字符串是不是以列表中的一个字符串结尾

Posted 2023-02-23

【中文标题】检查字符串是不是以列表中的一个字符串结尾

【英文标题】：Check if string ends with one of the strings from a list检查字符串是否以列表中的一个字符串结尾

【发布时间】：2013-08-23 12:02:38

【问题描述】：

以下代码的pythonic方式是什么？

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

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

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

【问题讨论】：

这个问题虽然很好回答，但或许作者原本想到的是if any((file_name.endswith(ext) for ext in extensions))。

【参考方案1】：

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

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

【讨论】：

你知道为什么它不接受列表而是接受元组吗？只是好奇

@falsetru 答案中的链接没有明确回答该问题。它只提到它可以 接受元组，但没有提到它不能 接受列表的原因。由于它们都是序列，我可能看到的唯一区别是列表是可变的，而元组是不可变的。我可能是错的，但我看不出有任何其他明确说明的原因。

如果要检查字符串是否以字母结尾：import string; str.endswith(tuple(string.ascii_lowercase))

请注意，endswith 仅接受 python 2.5 及以上版本的元组

从来不知道这个！太完美了！

【参考方案2】：

只需使用：

if file_name.endswith(tuple(extensions)):

【讨论】：

简单有效！

【参考方案3】：

有两种方法：正则表达式和字符串（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 次循环）

【参考方案4】：

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

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

【参考方案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

【讨论】：

你的意思是return filename.endswith(ext)？ :P

【参考方案7】：

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

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

【讨论】：

@Rainald62, index 在这种情况下应该是 rindex。