web-dev-qa-db-ja.com

文字列がリストの文字列のどちらかで終わっているかどうかを確認します。

以下のコードを書くためのPythonicの方法は何ですか？

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

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

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

pythonstringlist
190
TheMeaningfulEngineer

あまり知られていませんが、 str.endswith もTupleを受け入れます。あなたはループする必要はありません。

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

ただ使用してください：

if file_name.endswith(Tuple(extensions)):
38
Jon Clements

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

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

集合内の検索の時間の複雑さはO(1)（ docs ）であるため、集合を使用します。

5
alecxe

正規表現と文字列（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)

1ループあたり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)

1ループあたり274 ns±4.22 ns（7回の平均±標準偏差、それぞれ1000000ループ）

3
Igor A

何か他のものを探している間、私はちょうどこれに出会いました。

私は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.
1
Xxxo

もう1つの可能性は、INステートメントを利用することです。

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

一致する文字列のリストを返すことができる別の方法は

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

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

def has_extension(filename, extension):

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