详解Python 如何计算一个目录的大小

  • Post category:Python

计算一个目录的大小可以通过递归计算目录下所有文件的大小之和来实现。具体实现过程如下:

  1. 首先需要导入os模块,该模块提供了许多与文件系统相关的函数和方法。
  2. 编写一个函数,接收一个目录路径参数,递归遍历该目录下所有文件,计算它们的大小之和并返回。
  3. 在函数内部使用os.path.isdir()函数判断是否是一个目录,如果是目录则递归调用该函数,否则使用os.path.getsize()函数获取文件大小。
  4. 在递归过程中,使用一个变量累加目录下所有文件的大小。

下面是这个函数的实现示例:

import os

def get_dir_size(dir_path):
    size = 0
    for root, dirs, files in os.walk(dir_path):
        for f in files:
            size += os.path.getsize(os.path.join(root, f))
    return size

其中os.walk()函数用于递归遍历目录,返回三元组(root, dirs, files),其中root表示当前目录路径,dirs表示当前目录下的子目录列表,files表示当前目录下的文件列表。

接下来,我们通过两个示例来说明该函数的用法。

示例一:计算当前目录下所有文件的大小之和

print(get_dir_size('.'))
# 输出当前目录下所有文件的大小之和,单位是字节

示例二:计算指定目录下文件的大小之和

dir_path = 'C:/Users/username/Documents'
print(get_dir_size(dir_path))
# 输出指定目录下所有文件的大小之和,单位是字节

以上两个示例均输出目录大小的单位是字节,如果需要转换为其他单位,可以根据需要使用除法或乘法进行转换。