Python 按扩展名归类文件到文件夹：一键整理杂乱下载目录

用 Python 把文件夹里的文件按扩展名自动分到图片、文档、视频、压缩包等子文件夹，如 jpg/png 进「图片」、pdf/docx 进「文档」，附预览。

场景痛点

下载目录、桌面、微信接收文件夹永远是一团乱：图片、PDF、安装包、视频、表格混在一起，找个文件要翻半天。手动一个个拖到对应文件夹，分类规则一乱又得重排。用 Python 按扩展名自动归类，jpg/png 进「图片」、pdf/docx 进「文档」、mp4 进「视频」，没有匹配的进「其他」，一次整理干净。

用到的库

纯标准库，无需 pip 安装：

# os、shutil 为标准库

完整代码

import os
import shutil

# ========== 配置区 ==========
SOURCE_DIR = "./下载"        # 要整理的文件夹
DRY_RUN = True              # True=预览，False=真正移动
# 扩展名 -> 目标文件夹名（小写比较）
CATEGORY_MAP = {
"图片": [".jpg", ".jpeg", ".png", ".gif", ".bmp", ".webp", ".svg"],
"文档": [".pdf", ".doc", ".docx", ".xls", ".xlsx", ".ppt", ".pptx", ".txt", ".md"],
"视频": [".mp4", ".avi", ".mkv", ".mov", ".flv"],
"音频": [".mp3", ".wav", ".flac", ".aac"],
"压缩包": [".zip", ".rar", ".7z", ".tar", ".gz"],
"程序": [".exe", ".msi", ".dmg", ".apk"],
}
OTHER_DIR = "其他"
# ============================


def get_category(ext: str) -> str:
"""根据扩展名找分类，找不到返回 OTHER_DIR。"""
ext = ext.lower()
for folder, exts in CATEGORY_MAP.items():
if ext in exts:
return folder
return OTHER_DIR


def main():
if not os.path.isdir(SOURCE_DIR):
print(f"目录不存在：{SOURCE_DIR}")
return

count = 0
for name in os.listdir(SOURCE_DIR):
src_path = os.path.join(SOURCE_DIR, name)
# 只处理文件，跳过已经整理好的分类文件夹
if not os.path.isfile(src_path):
continue

ext = os.path.splitext(name)[1]
category = get_category(ext)
dest_dir = os.path.join(SOURCE_DIR, category)

# 目标重名时改名，避免覆盖
dest_path = os.path.join(dest_dir, name)
if os.path.exists(dest_path):
stem, e = os.path.splitext(name)
dest_path = os.path.join(dest_dir, f"{stem}_重名{e}")

action = "[预览]" if DRY_RUN else "[移动]"
print(f"{action} {name}  ->  {category}/")

if not DRY_RUN:
os.makedirs(dest_dir, exist_ok=True)   # 分类文件夹不存在就建
shutil.move(src_path, dest_path)
count += 1

print(f"\n共处理 {count} 个文件（DRY_RUN={DRY_RUN}）")


if __name__ == "__main__":
main()