文字列がリストの文字列のいずれかで終了しているかどうかを確認します

次のコードを書くPythonの方法は何ですか？

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

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

forループの明示的な宣言を回避してif条件に書き込むことができるという漠然とした記憶があります。これは本当ですか？

広く知られているわけではありませんが、str.endswithはタプルも受け入れます。ループする必要はありません。

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

ただ使う：

if file_name.endswith(tuple(extensions)):

ファイルから拡張子を取得し、それが拡張子のセットに含まれているかどうかを確認します。

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

セットでのルックアップの時間の複雑さがO（1）（docs）であるため、セットを使用します。

正規表現と文字列（str）メソッドの2つの方法があります。

通常、文字列メソッドの方が高速です（〜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ループ）

私はこれを持っています：

def has_extension(filename, extension):

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

私はこれに出くわしましたが、何か他のものを探していました。

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']