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")]