any() 测试中的语法错误

Question

我对 python 语法很陌生...在这种情况下如何使用 or 运算符？

为了检查文件是否以 '.txt' 或 '.sm3dump' 结尾，我尝试使用 any()，但出现 SyntaxError：

if any(filename.endswith('.txt'), filename.endswith('.sm3dump'))
                                                               ^
SyntaxError: invalid syntax

我做错了什么？

Answer 1

您使用两个参数调用了any()：

any(filename.endswith('.txt'), filename.endswith('.sm3dump'))
#   <----- argument #1 ----->  <------- argument #2 ------->

...但它只需要一个，它必须是可迭代的（列表、元组等）：

any([filename.endswith('.txt'), filename.endswith('.sm3dump')])
#   <-------------- one list with two elements -------------->

如果您想改用or，这很简单：

filename.endswith('.txt') or filename.endswith('.sm3dump')

使用any() 执行多个类似测试时的一个常见习惯用法是使用list comprehension 或generator expression：

any([filename.endswith(ext) for ext in ('.txt', '.sm3dump')])
#   <----------------- list comprehension ----------------->

any(filename.endswith(ext) for ext in ('.txt', '.sm3dump'))
#   <-------- generator expression (no brackets) -------->

虽然在这种情况下，您实际上可以向str.endswith() 提供一个字符串元组，它会检查它们：

filename.endswith(('.txt', '.sm3dump'))

无论您使用哪种测试，您都需要记住以冒号结束您的 if 语句：

if filename.endswith(('.txt', '.sm3dump')):
    # do something ...