【python】将不同后缀的文件分开到不同文件夹

功能

按照后缀将不同类型的文件区分，并保存到不同文件夹中。

举例

比如，原始的文件结构如下：

原始结构：
./dataset
└───img_file
│   │   a.jpg
│   │   b.jpg
│   │   c.png
│   │   d.png

其中包含jpg格式图像和png格式图像，将它们按照格式区分，拷贝到不同文件夹中：

目标结构：
./dataset
└───img_file
│   │   a.jpg
│   │   b.jpg
│   │   c.png
│   │   d.png
└───jpg_file
|   │   a.jpg
|   │   b.jpg
└───png_file
    │   c.png
    │   d.png

代码

import os
import shutil

def split_file(src, dst1, dst2):