温馨提示×

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

小亿
204
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