从子文件夹中提取文件

Question

我在下面有这段代码，仅从每个子文件夹中提取第一个文件。 如何修改以获取前 100 个文件？

import os

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

Answer 1

您可以从已排序的文件列表中分割前 100 个值，然后进行扩展。

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

Answer 2

用这个替换你的行

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

或者，由于files是一个列表，我们也可以这样做：

result_files += sorted(files)[:100]

这两项工作都是因为一个称为列表切片的 Python 特性。 您可以在此处和此处阅读有关它的更多信息。

另一种选择是这个：

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