Как проверить, содержит ли строка элемент из списка в Python

Question

У меня есть что-то вроде этого:

extensionsToCheck = ['.pdf', '.doc', '.xls']

for extension in extensionsToCheck:
    if extension in url_string:
        print(url_string)

Мне интересно, что было бы более элегантным способом сделать это в Python (без использования цикла for)? Я думал о чем-то вроде этого (например, из c / c ++), но это не сработало:

if ('.pdf' or '.doc' or '.xls') in url_string:
    print(url_string)

Редактировать: Я вынужден объяснить, чем это отличается от вопроса ниже, который помечен как потенциальный дубликат (поэтому он не закрывается, я думаю). Разница в том, что я хотел проверить, является ли строка частью некоторого списка строк, тогда как другой вопрос - проверка, является ли строка из списка строк подстрокой другой строки. Подобное, но не совсем то же самое и семантика имеют значение, когда вы ищете ответ в Интернете. Эти два вопроса на самом деле направлены на решение противоположной проблемы друг друга. Решение для обоих оказывается одинаковым.

Answer 1

Используйте генератор вместе с any, который закорачивает первый True:

if any(ext in url_string for ext in extensionsToCheck):
    print(url_string)

РЕДАКТИРОВАТЬ: Я вижу, этот ответ был принят OP. Хотя моё решение может быть «достаточно хорошим» решением его конкретной проблемы, и это хороший общий способ проверить, найдены ли какие-либо строки в списке в другой строке, имейте в виду, что это все, что делает это решение. Неважно, где находится строка, например в конце строки . Если это важно, как это часто бывает с URL-адресами, вам следует обратиться к ответу @Wladimir Palant, иначе вы рискуете получить ложные срабатывания.

Answer 2

extensionsToCheck = ('.pdf', '.doc', '.xls')

'test.doc'.endswith(extensionsToCheck)   # returns True

'test.jpg'.endswith(extensionsToCheck)   # returns False

Answer 3

Лучше правильно проанализировать URL - таким образом вы сможете правильно обрабатывать http://.../file.doc?foo и http://.../foo.doc/file.exe.

from urlparse import urlparse
import os
path = urlparse(url_string).path
ext = os.path.splitext(path)[1]
if ext in extensionsToCheck:
  print(url_string)

Answer 4

Используйте списочные выражения, если вы хотите однострочное решение. Следующий код возвращает список, содержащий строку url_string, если он имеет расширения .doc, .pdf и .xls, или возвращает пустой список, если он не содержит расширения.

print [url_string for extension in extensionsToCheck if(extension in url_string)]

ПРИМЕЧАНИЕ: Это только для проверки, содержит ли он или нет, и бесполезно, когда нужно извлечь точное слово, соответствующее расширениям.

Answer 5

Проверьте, соответствует ли это регулярному выражению:

'(\.pdf$|\.doc$|\.xls$)'

Примечание: если ваши расширения не в конце URL, удалите символы $, но это немного ослабит его

Answer 6

Это вариант ответа о понимании списка, заданного @ psun.

Переключая выходное значение, вы можете фактически извлечь шаблон сопоставления из понимания списка (что невозможно при подходе any()).by @ Lauritz-v-Thaulow)

extensionsToCheck = ['.pdf', '.doc', '.xls']
url_string = 'http://.../foo.doc'

print [extension for extension in extensionsToCheck if(extension in url_string)]

['. doc'] `

Кроме того, вы можете вставить регулярное выражение, если хотите собрать дополнительную информациюкак только сопоставленный шаблон известен (это может быть полезно, когда список разрешенных шаблонов слишком длинный для записи в один шаблон регулярного выражения)

print [re.search(r'(\w+)'+extension, url_string).group(0) for extension in extensionsToCheck if(extension in url_string)]

['foo.doc']