python计算目录中大小特定类型的文件txt

Question

在我的文件夹中，有两种类型的文件:html 和txt。我想知道 txt 文件的总大小。

我找到了这段代码，但我该如何应用它来满足我的需要？

import os
from os.path import join, getsize
size = 0
count = 0
for root, dirs, files in os.walk(path):
    size += sum(getsize(join(root, name)) for name in files)
    count += len(files)
print count, size

Answer 1

您可以通过向理解中添加 if 来限定哪些文件，例如:

for root, dirs, files in os.walk(path):
    size += sum(getsize(join(root, name)) for name in files if name.endswith('.txt'))
    count += sum(1 for name in files if name.endswith('.txt'))
print count, size