有一批MRI图像数据,只想保留期中既有6月份也有12月份的数据
implementation
"""
Author:tx
time:2021-08-09
"""
import glob
import os
import shutil
old_6m = r"C:\Users\tx\Desktop\MRIdata\6_month"
new_6m =r"C:\Users\tx\Desktop\MRIdata\new6"
old_12m = r"C:\Users\tx\Desktop\MRIdata\12_month"
new_12m = r"C:\Users\tx\Desktop\MRIdata\new12"
#记录6月 跟12月 所有文件名
Fof6 = os.listdir(old_6m)
Fof12 = os.listdir(old_12m)
#遍历6月跟12月 若有重合的ID则将其分别复制到新文件夹中
# 复制文件夹函数
def CP2(sourcepath,targetpath):
shutil.copytree(sourcepath,targetpath)
#判断6月文件夹名是否有与12月重合ID 如果有则返回相同ID的12月文件
def Issame(filename,Fof12):
for i in Fof12:
if i.split('_')[2] == filename.split('_')[2]:
return i
#遍历6月所有文件名
for i in Fof6:
#判断是否存在相同的文件名
if Issame(i,Fof12):
#如果存在相同文件名字分别将两个文件移动到目标文件
#首先获得6月sourcepath跟targetpath
sp6 = os.path.join(old_6m,i)
tg6 = os.path.join(new_6m,i)
#复制文件
CP2(sp6,tg6)
#12月类似
sp12 = os.path.join(old_12m,Issame(i,Fof12))
tg12 = os.path.join(new_12m,Issame(i,Fof12))
CP2(sp12,tg12)