用 Python 快速找到最大的文件

somenzz

发布于 2021-11-12 08:41:04

71900

代码可运行

文章被收录于专栏：Python七号Python七号

运行总次数：0

代码可运行

现在的电脑差不多都是固态硬盘了，速度很快，但容量不会太大，经常会出现磁盘空间不足的情况，怎么办，删除那些不重要的最大的文件是最有效的办法。

那么如何用找到最大的文件呢？比如说找到最大的前 10 个文件？

思路：我们遍历目录，将文件路径和文件大小作为生成器返回，然后插入大小为 10 的大顶堆，最后将大顶堆的内容打印即可。

借助 Python，代码很简洁：

import os
import time

from os.path import join, getsize
from heapq import nlargest


def walk_files_and_sizes(start_at: str):
    for root, _, files in os.walk(start_at):
        for file in files:
            path = join(root, file)
            try:
                size = getsize(path)  # bytes
                yield path, size
            except OSError:
                continue


def largest_files(n: int, start_at: str) -> None:
    MB = 1024 * 1024
    largest = nlargest(n, walk_files_and_sizes(start_at), key=lambda x: x[1])

    for path, size in largest:
        print(f'{size//MB} MB {path}')



if __name__ == '__main__':
    start = time.perf_counter()
    largest_files(10, "/Users/aaron/")
    elapsed = time.perf_counter() - start
    print(f'{elapsed} seconds elapsed')