os.listdir() 不打印出所有文件

Question

I've got a bunch of files and a few folders. I'm trying to append the zips to a list so I can extract those files in other part of the code. It never finds the zips.

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

The path is fine or it wouldn't print out anything. It picks up the same files each time but will not pick up any others. It picks up about 1/5th of the files in the directory.

At a complete loss. I've made sure that some weird race condition with the file availability isn't the problem by putting a time.sleep(3) in the code. Didn't solve it.

Answer 1

It's possible your files have more than one period in them. Try using str.endswith :

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

Another good idea (thanks, Jean-François Fabre!) is to useos.path.splitext , which handles the extension quite nicely:

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

Am even better solution, I recommend with the glob.glob function:

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

Answer 2

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