web-dev-qa-db-fra.com

str.startswith avec une liste de chaînes à tester

J'essaie d'éviter d'utiliser autant d'instructions if et de comparaisons et d'utiliser simplement une liste, mais je ne sais pas comment l'utiliser avec 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"

Ce que j'aimerais que ce soit, c'est:

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

Toute aide serait appréciée.

pythonstringlist
142
Eternity

str.startswith vous permet de fournir un tuple de chaînes à tester:

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

De la docs :

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

Renvoie True si la chaîne commence par prefix, sinon renvoie False. prefix peut également être un tuple de préfixes à rechercher.

Ci-dessous une démonstration:

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

Vous pouvez également utiliser any() , map() comme suit:

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

Ou alternativement, en utilisant compréhension de liste :

if any([l.startswith(s) for s in x])
    pass # Do something
19
user764357