将文件进行分类整理
功能
根据一个文件夹中的文件类型建立相应的文件夹,将同一种类型的文件放在一个文件夹中。
实现思路
主要用到 os 和 shutil 两个库,os 用来获取文件夹中的文件对象、创建文件夹,shutil 用来移动文件到指定文件夹中。
使用方法
当然,运行需要 python 环境,并且需要安装 os 和 shutil 两个库。
由于代码中用的是相对路径,所以需要把代码文件放到要处理文件的文件夹中。
代码
import shutil
import os
path = './'
files = os.listdir(path)
for f in files:
folder_name = './' + f.split('.')[-1]
if not os.path.exists(folder_name):
os.makedirs(folder_name)
shutil.move(f,folder_name)
else:
shutil.move(f,folder_name)