功能
按照后缀将不同类型的文件区分,并保存到不同文件夹中。
举例
比如,原始的文件结构如下:
原始结构:
./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):