str.startswith со списком строк для проверки


183

Я пытаюсь избежать использования очень многих операторов if и сравнений и просто использовать список, но не уверен, как его использовать с str.startswith:

if link.lower().startswith("js/") or link.lower().startswith("catalog/") or link.lower().startswith("script/") or link.lower().startswith("scripts/") or link.lower().startswith("katalog/"):
    # then "do something"

То, что я хотел бы, чтобы это было:

if link.lower().startswith() in ["js","catalog","script","scripts","katalog"]:
    # then "do something"

Любая помощь будет оценена.


7
Ответы на этот вопрос напоминают мне, почему я так люблю Python!
dotancohen

Даже если по какой-то причине вам необходимо выполнить несколько проверок, вы должны сохранить результат ˋlower () ˋ. Нет смысла делать это снова и снова.
Бахсау

Ответы:


346

str.startswith позволяет предоставить набор строк для проверки:

if link.lower().startswith(("js", "catalog", "script", "katalog")):

Из документов :

str.startswith(prefix[, start[, end]])

Возврат, Trueесли строка начинается с prefix, в противном случае возврат False. prefixтакже может быть кортеж префиксов для поиска.

Ниже приведена демонстрация:

>>> "abcde".startswith(("xyz", "abc"))
True
>>> prefixes = ["xyz", "abc"]
>>> "abcde".startswith(tuple(prefixes)) # You must use a tuple though
True
>>>

3
И если он не поддерживает это, вы все равно можете сделать это с anyпомощью genxp.
user2357112 поддерживает Monica

Условно-условное, а не «возможно, ОП на старой версии».
user2357112 поддерживает Monica

эй, если я использую это как часть функции, как я могу заставить его вернуть элемент из списка, которому он соответствует?
roastbeeef

@roastbeeef «для цикла». Других вариантов нет. ;)
Бахсау

26

Вы также можете использовать any(), map()например, так:

if any(map(l.startswith, x)):
    pass # Do something

Или альтернативно, используя выражение генератора :

if any(l.startswith(s) for s in x)
    pass # Do something

6
Не используйте понимание списка с любым; использовать генератор.
DylanYoung
Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.