Question

我试图避免使用如此多的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"

任何帮助都将不胜感激。

Answer 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
>>>

Answer 2

你也可以这样使用any()，map()：

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

或者，使用list comprehension：

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

str.startswith，带有要测试的字符串列表

2 个答案: