Überprüfen Sie, ob die Zeichenfolge mit einer der Zeichenfolgen aus einer Liste endet

Wie schreibt man den folgenden Code pythonisch?

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

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

Ich habe eine vage Erinnerung daran, dass die explizite Deklaration der forSchleife vermieden und in die ifBedingung geschrieben werden kann. Ist das wahr?

Obwohl nicht allgemein bekannt, akzeptiert str.endswith auch ein Tupel. Sie müssen nicht schleifen.

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

Benutz einfach:

if file_name.endswith(tuple(extensions)):

Nehmen Sie eine Erweiterung aus der Datei und prüfen Sie, ob sie in den Erweiterungen enthalten ist:

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

Verwenden eines Satzes, da die zeitliche Komplexität für die Suche in Sätzen O (1) ( docs ) beträgt .

Es gibt zwei Möglichkeiten: reguläre Ausdrücke und String (str) -Methoden.

String-Methoden sind normalerweise schneller (~ 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 pro Schleife (Mittelwert ± Standardabweichung von 7 Läufen, jeweils 1000000 Schleifen)

file_name = 'test.mp3'
extensions = ('.mp3','.avi')
print(file_name.lower().endswith(extensions))
%timeit file_name.lower().endswith(extensions)

274 ns ± 4,22 ns pro Schleife (Mittelwert ± Standardabweichung von 7 Läufen, jeweils 1000000 Schleifen)

Ich habe das:

def has_extension(filename, extension):

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

Ich bin gerade darauf gestoßen, als ich nach etwas anderem gesucht habe.

Ich würde empfehlen, mit den Methoden im osPaket zu gehen . Dies liegt daran, dass Sie es allgemeiner gestalten können, um jeden seltsamen Fall zu kompensieren.

Sie können so etwas tun wie:

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.

Eine andere Möglichkeit könnte darin bestehen, die IN-Anweisung zu verwenden:

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

Eine andere Möglichkeit, die Liste der übereinstimmenden Zeichenfolgen zurückzugeben, ist

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