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

我刚看到这个，同时寻找其他东西。

我建议您使用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 5

我有这个：

def has_extension(filename, extension):

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

Answer 6

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

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

Answer 7

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

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

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

7 个答案: