用python写个文件分类的代码

以下是一个简单的文件分类代码，可以将指定目录下的文件按照类型分类并移动到对应的文件夹中：

import os 
import shutil 

# 定义文件类型和对应的文件夹名称 
filetypes = {
    '图片': ['.jpg', '.jpeg', '.png', '.gif', '.bmp'],
    '文档': ['.doc', '.docx', '.txt', '.pdf', '.xlsx', '.pptx'],
    '音乐': ['.mp3', '.wav', '.flac'],
    '视频': ['.mp4', '.avi', '.mov', '.wmv'],
    '程序': ['.py', '.exe', '.sh', '.bat'],
    '压缩包': ['.zip', '.rar', '.7z']
}

# 指定待分类的目录 
source_dir = 'C:/Users/UserName/Desktop/'

# 遍历目录下的所有文件 
for filename in os.listdir(source_dir):
    # 获取文件的完整路径 
    file_path = os.path.join(source_dir, filename)
    # 判断是否为文件 
    if os.path.isfile(file_path):
        # 获取文件扩展名 
        file_ext = os.path.splitext(filename)[1]
        # 遍历文件类型字典，查找对应的文件夹 
        for filetype, ext_list in filetypes.items():
            if file_ext in ext_list:
                # 创建对应的文件夹 
                dest_dir = os.path.join(source_dir, filetype)
                if not os.path.exists(dest_dir):
                    os.makedirs(dest_dir)
                # 移动文件到对应的文件夹中 
                shutil.move(file_path, os.path.join(dest_dir, filename))
                break

在上面的代码中，我们首先定义了一个文件类型字典，包含了各种文件类型和对应的文件扩展名。然后指定了待分类的目录，并遍历目录下的所有文件。对于每个文件，我们获取其扩展名并遍历文件类型字典，查找对应的文件夹。如果找到了对应的文件夹，就创建该文件夹（如果不存在），然后将文件移动到该文件夹中。最后，我们就可以得到一个按照文件类型分类的目录结构。