欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

用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对相同文件类型,不同文件名的文件进行分类保存。
文件名如下图所示:
用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

相关标签: python