批量修改txt中图片的文件名，并提取到另一文件夹

import shutil
import os
import re
import numpy as np
data = []
#FLIR_09593_PreviewData
s = 'PreviewData'
r='RGB'
for line in open("/media/qq/work/flir/data/ImageSets/train.txt", "r"):
    #line
    # 'FLIR_09593_PreviewData
    #_RGB
    #if s in fileName:
    cs=line.replace(s, r)
    #os.rename(line, line.replace(s, r))
    data.append(cs) # print(data) # data中得到了txt文件的所有图像的名称
for a in data:
    srcfile_path = '/media/qq/work/flir/data/JPEGImages/{}'.format(a[:-1]) + '.jpg' #因为我这个txt中的图像名称没有后缀，故添加一个
    tarfile_path = '/media/qq/work/flir/data/infrared/train/' #新创建一个文件夹
    shutil.copy(srcfile_path, tarfile_path)