python - os.listdir() 没有打印出所有文件

Question

我有一堆文件和几个文件夹。我正在尝试将 zip 附加到列表中，以便可以在代码的其他部分中提取这些文件。它永远找不到拉链。

for file in os.listdir(path):
     print(file)
     if file.split(".")[1] == 'zip':
     reg_zips.append(file)

路径很好，否则它不会打印出任何东西。它每次都拾取相同的文件，但不会拾取任何其他文件。它提取了目录中大约 1/5 的文件。

一败涂地。通过在代码中放置 time.sleep(3)，我确保文件可用性的一些奇怪的竞争条件不是问题。没解决。

Answer 1

您的文件中可能包含多个句点。尝试使用str.endswith：

reg_zips = []
for file in os.listdir(path):
     if file.endswith('zip'):
         reg_zips.append(file)

另一个好主意（感谢 Jean-François Fabre！）是使用os.path.splitext，它可以很好地处理扩展：

if os.path.splitext(file)[-1] == '.zip':
    ... 

甚至更好的解决方案，我建议使用以下glob.glob功能：

import glob
reg_zips = glob.glob('*.zip')

Answer 2

reg_zips = [z for z in os.listdir(path) if z.endswith("zip")]