Str.startswith со списком строк для тестирования
Я пытаюсь избежать использования так много операторов 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"
Любая помощь будет оценена.
Ответы
Ответ 1
str.startswith
позволяет вам снабжать кортеж строк для проверки:
if link.lower().startswith(("js", "catalog", "script", "katalog")):
Из docs:
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
>>>
Ответ 2
Вы также можете использовать any()
, map()
вот так:
if any(map(l.startswith, x)):
pass # Do something
Или, альтернативно, используя понимание списка:
if any([l.startswith(s) for s in x])
pass # Do something