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

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'))