从子文件夹中提取文件

Question

I have this code below extract just the first file from each subfolder. How do I modify to get say the first 100 files?

import os

result_files = []
for root, dirs, files in os.walk(your_folder):
    if files:
        result_files.append(sorted(files)[0])

Answer 1

You can slice the first 100 values from the sorted list of files and then extend.

result_files.extend(sorted(files)[:100])

Answer 2

Replace the line you had with this

result_files.extend(sorted(files)[:100])

Or, since files is a list, we can also do:

result_files += sorted(files)[:100]

Both of these work because of a pythonic feature called list slicing. You can read more about it here and here .

Another option is this one:

for i in range(100): result_files.append(sorted(files)[i])