python - 查找目录中所有文件的总文件大小

Question

目前我正在使用下面的代码来计算目录中所有文件的大小。这很适合我的使用。只是想知道是否有办法让它获取特定文件类型(例如 .py 文件或 .txt)的总文件大小。

total_file = sum(os.path.getsize(f) for f in os.listdir('.') if os.path.isfile(f))

Answer 1

使用str.endswith('.py') (用于过滤 .py 文件):

total_file = sum(os.path.getsize(f) for f in os.listdir('.')
    if os.path.isfile(f) and f.endswith('.py'))