python - 从文件夹中选择前 n 个最小的文件

Question

我有一个包含很多文件的目录，大约 50k。我想选择 n 个最小的文件。

import os

def main():
listfile = os.listdir("../../folder")
for singlefile in listfile:
    print singlefile

使用这段代码，不幸的是，我将全部选中它们。

Answer 1

您可以使用 os.path.getsize获取文件大小，然后根据它对文件进行排序(heapq.nsmallest 通常是首选，但已经包含在另一个答案中，所以我将展示一个替代方案):

import os

directory = "../../folder"
n = 10

files = os.listdir(directory)
files_sorted_by_size = sorted(files, key=lambda filename: os.path.getsize(os.path.join(directory, filename)))  
nsmallest_files = files_sorted_by_size[:n] 
print(nsmallest_files)  

一行:

sorted(os.listdir(directory), key=lambda filename: os.path.getsize(os.path.join(directory, filename)))[:n]