平时桌面或文件夹内鱼龙混杂，各种类型的文件都有怎么办？
本篇文章中，我们将学习如何使用 Python 编写一个文件整理分类的脚本。
该脚本能够自动获取文件类型，并将文件按照类型整理到不同的子文件夹中。
先看下效果，整理前各种类型文件都有：
整理后，根据文件类型放置到了不同的文件夹中
那下面正式开始上代码：
1. 安装必要的库
首先，确保你已经安装了 shutil 和 os 库。如果没有安装，可以使用以下命令进行安装：
pip install shutil
2. 编写整理分类脚本
下面是文件整理分类脚本的完整代码，可直接使用，输入需要整理的文件夹路径即可：
import os
import shutil
def get_file_type(file_path):
# 获取文件扩展名
_, file_extension = os.path.splitext(file_path)
return file_extension.lower()
def organize_files(input_folder):
# 获取输入文件夹中的所有文件
files = os.listdir(input_folder)
# 遍历每个文件
for file in files:
# 构建文件的完整路径
file_path = os.path.join(input_folder, file)
# 判断是否是文件
if os.path.isfile(file_path):
# 获取文件类型
file_type = get_file_type(file_path)
# 构建目标文件夹路径
target_folder = os.path.join(input_folder, file_type + '类')
# 如果目标文件夹不存在，创建它
if not os.path.exists(target_folder):
os.makedirs(target_folder)
# 构建目标文件路径
target_file_path = os.path.join(target_folder, file)
# 移动文件到目标文件夹
shutil.move(file_path, target_file_path)
# 使用示例
organize_files('files')
3. 代码详解
get_file_type 函数
def get_file_type(file_path):
# 获取文件扩展名
_, file_extension = os.path.splitext(file_path)
return file_extension.lower()
这个函数接受文件路径作为参数，使用 os.path.splitext 获取文件的扩展名（例如 .pdf、.jpg），并返回小写的扩展名。
organize_files 函数
def organize_files(input_folder):
# 获取输入文件夹中的所有文件
files = os.listdir(input_folder)
# 遍历每个文件
for file in files:
# 构建文件的完整路径
file_path = os.path.join(input_folder, file)
# 判断是否是文件
if os.path.isfile(file_path):
# 获取文件类型
file_type = get_file_type(file_path)
# 构建目标文件夹路径
target_folder = os.path.join(input_folder, file_type + '类')
# 如果目标文件夹不存在，创建它
if not os.path.exists(target_folder):
os.makedirs(target_folder)
# 构建目标文件路径
target_file_path = os.path.join(target_folder, file)
# 移动文件到目标文件夹
shutil.move(file_path, target_file_path)
这个函数接受一个输入文件夹路径，使用 os.listdir 获取该文件夹中的所有文件。
然后，遍历每个文件，获取文件类型并自动创建对应的文件夹，最后将文件移动到相应的文件夹中。
4. 运行脚本
保存以上代码到一个文件（例如 organize_files.py），然后在终端或者编辑器中运行脚本：
python organize_files.py
脚本将会处理指定文件夹中的文件，按照文件类型分类整理。
通过这个文件整理分类脚本，你可以方便地将一个混杂的文件夹整理成按照类型分类的结构。
有所收益，欢迎关注，持续更新，一起进步~
