python之遍历文件夹统计所有文件大小递归及循环两种解法

off999 2024-10-15 12:10 45 浏览 0 评论

统计一个目录下所有文件总计大小难点在遍历所有子目录，以下列出2种解法。

递归解

递归解题的思路没什么出奇的，代码特别简洁易懂。遍历一个目录，累计所有文件的大小，对于子目录则调用自身递归统计子目录。

需要注意的有两点：

一、路径书写强烈建议用“/”，这种写法在windows系统和linux系统通用。

二、文件与路径拼接方法，经time测试用f字符串拼接方式比用os.path.join()方式拼接效率略高一些，而且书写也简洁一些。

import os


def getdir1(path):
    name_list = os.listdir(path)
    size = 0
    for i in name_list:
        j = f'{path}/{i}'  # 文件和路径拼接用字符串+更直接更高效
        # j = os.path.join(path,i)
        if os.path.isdir(j):
            size += getdir1(j)
        else:
            size += os.path.getsize(j)
    return size


print(getdir1('D:/学习资料'))

out:
304372584432

循环解

循环解题的思路要注意以下两点：

一、对列表的pop和append用法，这样实际是将列表当成了堆栈使用，后进先出。

二、注意 while name_dir的写法，这个是判断name_dir列表是否为空，为空则结束循环，不为空则循环。

def getdir2(path):
    size = 0
    name_dir = [path]
    while name_dir:
        name_list = os.listdir(ret := name_dir.pop())
        for i in name_list:
            j = f'{ret}/{i}'
            if os.path.isfile(j):
                size += os.path.getsize(j)
            else:
                name_dir.append(j)
    return size


print(getdir2('D:/学习资料'))

out:

python获取文件大小