统计一个目录下所有文件总计大小难点在遍历所有子目录,以下列出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:
304372584432