Python如何正确获取目录下后缀为json或xlsm的所有文件列表
错误原因
- 原代码的逻辑判断写法不符合Python运算规则:
(".json" or ".xlsm")中or是布尔运算符,非空字符串的布尔值为True,or运算会直接返回第一个为真的操作数,也就是".json",所以整个判断等价于只检查文件名是否包含.json,自然只能筛选出json后缀的文件。 - 移除括号后,表达式变为
".json" or ".xlsm" in s,由于in运算符优先级高于or,实际运算逻辑为".json" or (".xlsm" in s),第一个值.json恒为真,所以整个判断永远为True,会返回目录下所有文件。 - 把
or替换为and后,判断逻辑变为(".json" and ".xlsm") in s,and运算在左右操作数都为真时返回第二个操作数也就是".xlsm",等价于只判断文件名是否包含.xlsm,所以只会返回xlsm后缀的文件,刚好对应你遇到的异常现象。
修正方案
str.endswith()方法本身支持传入元组批量匹配多个后缀,是实现该需求最简洁的写法:
import os filelist = os.listdir(path='.') target_list = [s for s in filelist if s.endswith(('.json', '.xlsm'))]
如果要拆分判断写也可以,效果完全一致:
import os filelist = os.listdir(path='.') target_list = [s for s in filelist if s.endswith('.json') or s.endswith('.xlsm')]
内容的提问来源于stack exchange,提问作者Piotr Chmurzynski
相关产品推荐
相关产品推荐

