用Python按照文件名对文件分类并保存
程序员文章站
2022-06-26 17:43:33
使用Python对相同文件类型,不同文件名的文件进行分类保存。文件名如下图所示:按照上图中 ’ _ '下划线后文件名(ct,seg)进行分类并保存到指定文件夹中。代码如下:import osimport shutilpath = r'./Train'RawData = r'./RawData'LabelData = r'./LabelData'files_list = os.listdir(path)for file in files_list: filename, s...
使用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
...
本文地址:https://blog.csdn.net/weixin_43380735/article/details/109629245
上一篇: 编程语法之 类,对象,属性,使用方法