写一个python 根据文件名进行分类
时间: 2024-02-01 15:03:54 浏览: 34
好的,我可以回答这个问题。你可以使用 Python 的 os 模块来获取文件名,并根据一些规则进行分类。示例代码如下:
```
import os
# 定义文件分类的规则
rules = {
'image': ['.jpg', '.png', '.gif'],
'text': ['.txt', '.doc', '.pdf'],
'video': ['.mp4', '.avi', '.mov']
}
# 获取目标文件夹中的所有文件
files = os.listdir('/path/to/files')
# 根据文件名进行分类
file_types = {}
for file in files:
# 获取文件名和扩展名
file_name, file_ext = os.path.splitext(file)
# 判断文件类型并添加到对应的列表中
for type_name, ext_list in rules.items():
if file_ext in ext_list:
if type_name not in file_types:
file_types[type_name] = []
file_types[type_name].append(file)
# 输出分类结果
for type_name, file_list in file_types.items():
print(type_name, len(file_list))
```
这个代码示例将文件分为了三类:图像、文本和视频,根据文件扩展名来判断文件类型,并将同一类型的文件放在同一个列表中。你可以根据具体的需求来修改分类规则。