Python - 获取所有目录的文件扩展名

Question

我们有一个包含数十万个文件的硬盘

我需要弄清楚我们有多少个文件扩展名

我怎样才能用 python 做到这一点？

我需要它遍历每个目录。我公司的律师需要这个。它可以是整个硬盘驱动器的总数它不必按目录分割

例子:

1232 JPEG
11 exe
45 bat
2342 avi
532 doc

Answer 1

查看 os 模块中的 os.walk 调用并遍历整个目录树。使用 os.path.splitext 获取扩展。维护一个字典，其中键为 extension.lower() 并增加您遇到的每个扩展的计数。

import os
import collections
extensions = collections.defaultdict(int)

for path, dirs, files in os.walk('/'):
   for filename in files:
       extensions[os.path.splitext(filename)[1].lower()] += 1

for key,value in extensions.items():
    print 'Extension: ', key, ' ', value, ' items'