1 import os
 2 
 3 # 确认需要整理的文件类型
 4 file_types = ['txt','jpg','exe','py']
 5 
 6 # 根据文件类型创建相应的文件夹
 7 for item in file_types:
 8     os.mkdir(item)
 9 
10 # 遍历桌面文件并将其移动到对应的文件夹下
11 for file_name in os.listdir('./'):
12     for file_type in file_types:
13         if file_name.endswith('.'+file_type):
14             # 将文件移动到同名文件夹中
15             os.rename(file_name,file_type + '/' + file_name)