Проверьте, заканчивается ли строка одной из строк из списка

220

Каков питонный способ написания следующего кода?

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

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

У меня есть смутное воспоминание, что явного объявления forцикла можно избежать и записать в ifусловии. Это правда?

python string list TheMeaningfulEngineer
источник
2
Хотя на этот вопрос хорошо ответили, возможно, автор изначально задумался if any((file_name.endswith(ext) for ext in extensions)).
Сафт

Ответы:

450

Хотя str.endswith не очень широко известен, он также принимает кортеж. Вам не нужно зацикливаться.

>>> 'test.mp3'.endswith(('.mp3', '.avi'))
True
falsetru
источник
10
Вы знаете, почему он не принимает список, но делает кортеж? просто любопытно
ilyail3
2
@falsetru Ссылка в ответе не дает четкого ответа на этот вопрос. В нем только упоминается, что он может принимать кортежи, но не почему он не может принимать списки. Поскольку они обе являются последовательностями, единственное различие, которое я могу видеть, состоит в том, что списки являются изменяемыми, а кортежи неизменными. Я могу ошибаться, но я не вижу никакой другой причины, по которой это прямо указано.
KymikoLoco
4
Если вы хотите проверить, заканчивается ли строка буквой:import string; str.endswith(tuple(string.ascii_lowercase))
Алекс Виллисон
3
только примечание, endswithпринимает кортеж только для Python 2.5 и выше
Акаш Сингх
1
Никогда этого не знал! Отлично!
fool4jesus
46

Просто используйте:

if file_name.endswith(tuple(extensions)):
Джон Клементс
источник
Просто и эффективно!
Ced
6

Возьмите расширение из файла и посмотрите, есть ли оно в наборе расширений:

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

Использование набора, потому что сложность времени для поиска в наборах составляет O (1) ( документы ).

alecxe
источник
8
Просто отметим, что, как вы упомянули, эффективность для довольно коротких кортежей .endswith()с интернированным кортежем будет быстрее, чем поиск по множеству
Джон Клементс
@JonClements Я думаю, вам нужен особый золотой значок комментария SO, чтобы делать отличные заметки об ответах и ​​вопросах :)
alecxe
Нет, я просто иду за значком "Подглядывающий алекс";)
Джон Клементс
2
Также обратите внимание, что в версии 2.7 и новее вы можете использовать математический синтаксис для множеств, {'.mp3','.avi'}это позволяет избежать дополнительного преобразования типов и может быть более читабельным в зависимости от вашего фона («Хотя это может вызвать путаницу со словарями и не может быть использовано для создания пустых»). наборы).
Перкинс
@JonClements когда-нибудь я стану таким же мудрым, как ты :)
alecxe
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 нс ± 1,83 нс на цикл (среднее ± стандартное отклонение из 7 циклов, 1000000 циклов в каждом)

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

274 нс ± 4,22 нс на цикл (среднее ± стандартное отклонение 7 циклов, 1000000 циклов в каждом)

Игорь А
источник
1

У меня есть это:

def has_extension(filename, extension):

    ext = "." + extension
    if filename.endswith(ext):
        return True
    else:
        return False
Томас Воутерс
источник
1
Вы имеете в виду return filename.endswith(ext)? : P
Mr_and_Mrs_D
1

Я только сталкивался с этим, ища что-то еще.

Я бы порекомендовал пойти с методами в 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.
Xxxo
источник
0

Другой возможностью может быть использование оператора IN:

extensions = ['.mp3','.avi']
file_name  = 'test.mp3'
if "." in file_name and file_name[file_name.rindex("."):] in extensions:
    print(True)
NeverHopeless
источник
@ Rainald62, indexдолжно быть rindexв этом случае.
NeverHopeless
0

Другой способ, который может вернуть список совпадающих строк: 

sample = "alexis has the control"
matched_strings = filter(sample.endswith, ["trol", "ol", "troll"])
print matched_strings
['trol', 'ol']
Акаш Сингх
источник