温馨提示×

怎么用python统计文件类型及数量

小亿
99
2023-10-21 11:00:14
栏目: 编程语言

要统计文件类型及数量,你可以使用Python的os模块来遍历文件夹中的文件,然后使用文件的后缀来判断文件类型。以下是一个示例代码:

import os

def count_file_types(folder_path):
    file_types = {}

    for root, dirs, files in os.walk(folder_path):
        for file in files:
            file_name, file_ext = os.path.splitext(file)
            if file_ext:
                file_type = file_ext[1:].lower()
                if file_type in file_types:
                    file_types[file_type] += 1
                else:
                    file_types[file_type] = 1

    return file_types

folder_path = 'path/to/folder'
file_types = count_file_types(folder_path)

for file_type, count in file_types.items():
    print(f"{file_type}: {count}")

在上述示例代码中,count_file_types函数接受一个文件夹路径作为参数,使用os.walk遍历该文件夹中的所有文件。对于每个文件,使用os.path.splitext函数来获取文件的文件名和后缀。然后,将后缀转换为小写,并使用字典file_types来记录每个文件类型的数量。最后,遍历file_types字典,输出每个文件类型及其对应的数量。

请注意,上述代码只会统计文件的后缀作为文件类型。如果你需要更精确的文件类型信息,可以使用第三方库如mimetypes来识别文件的MIME类型。

0