如何找到目录及其所有子目录中所有文件的扩展名？

Question

我试图列出目录及其所有子目录中所有文件的扩展名（即类型）。 我已成功列出所有文件名，这是我的代码：

import os
path = os.getcwd()
for dirpath, dirnames, filenames in os.walk(path):
    for file in filenames:
        file_path = os.path.join(dirpath, file)
        file_size = os.path.getsize(file_path)
        print("{} : {}".format(file_path, round(file_size, 3)))
    for dirname in dirnames:
        dir_path = os.path.join(dirpath, dirname)
        dir_size = os.path.getsize(dir_path)
        print("{} : {}".format(dir_path, round(dir_size, 3)))

我想获取目录及其所有子目录中所有文件的扩展名（即类型）。 任何帮助将不胜感激！

Answer 1

这还将为您提供目录及其子目录中的唯一扩展名

import os
path = os.getcwd()
extensions = set()

for dirpath, dirnames, filenames in os.walk(path):
    for file in filenames:
        file_extension = os.path.splitext(file)[1]
        extensions.add(file_extension)

print("Extensions found:", extensions)

Answer 2

以下是使用正则表达式提取文件扩展名的方法。

\.([^.]*)$表示“在字符串的末尾，匹配一个点后跟任何不是点的东西，并将非点部分保存到结果中”。

请注意，如果文件没有扩展名 - 例如 Windows hosts文件 - 匹配将为None ，在这种情况下我们需要将其替换为空字符串。

import os
import re

prog = re.compile(r'\.([^.]*)$')

path = os.getcwd()

for dirpath, dirnames, filenames in os.walk(path):
    for file in filenames:
        file_path = os.path.join(dirpath, file)
        m = prog.search(file)
        file_ext = m.group(1) if m else ''
        file_size = os.path.getsize(file_path)
        print("{} : {} : {}".format(file_path, file_ext, round(file_size, 3)))