用Python按照文件名对文件分类并保存

用Python按照文件名对文件分类并保存

使用Python对相同文件类型，不同文件名的文件进行分类保存。
文件名如下图所示：

按照上图中 ’ _ '下划线后文件名(ct，seg)进行分类并保存到指定文件夹中。
代码如下：

import os
import shutil

path = r'./Train'
RawData = r'./RawData'
LabelData = r'./LabelData'
files_list = os.listdir(path)

for file in files_list:
    filename, suffix = os.path.splitext(file)  # filename是文件名 suffix是文件后缀
    file_path = path + "/" + filename + '.gz'
    label1 = filename.split('_')[1]  # '_'后面的文件名
    label2 = label1.split('.')[0]  # '.'前面的文件名
    if label2 == 'ct':
        shutil.copy(file_path, RawData)
    else:
        shutil.copy(file_path, LabelData)

# 打印出的结果filename  label1  label2  依次为：
volume-covid19-A-0003_ct.nii
ct.nii
ct
volume-covid19-A-0003_seg.nii
seg.nii
seg
...