运用python为不平衡数据上采样

训练数据train.list如下：

E:/MyCode/fushikang2019/ImgOCT/normal/171221123612_00002/s077_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/normal/171225150539_00002/s062_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/normal/171222103742_00012/s173_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/normal/180103103434_00002/s093_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/abnor/171221162036_00011/s130_abnor.jpg
E:/MyCode/fushikang2019/ImgOCT/normal/171221093818_00011/s139_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/abnor/171223163916_00002/s010_abnor.jpg
E:/MyCode/fushikang2019/ImgOCT/normal/180102145143_00011/s195_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/normal/171229094108_00011/s136_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/normal/171223100609_00002/s029_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/normal/171229094022_00011/s011_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/normal/171221141331_00009/s036_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/abnor/171222103915_00011/s179_abnor.jpg
E:/MyCode/fushikang2019/ImgOCT/normal/171221093818_00011/s072_normal.jpg
E:/MyCode/fushikang2019/ImgOCT/abnor/171229154214_00011/s065_abnor.jpg

1.将文件中各个类别文件（normal，abnor）解析出来，
并将各个类别文件列表分别输出到不同文件，

输出前较少类别文件乱序并拷贝三份

#将文件中各个类别文件解析出来，
#并将各个类别文件列表分别输出到不同文件，输出前较少类别文件乱序并拷贝三份
import os
import random
mf = open('train1.list','r')
aa = []
nn = []
for line in mf:
   # print(line)
    if 'abnor' in line:
       # print(line)
        aa.append(line)
    if 'normal' in line:
        #print(line)
        nn.append(line)
mf.close()
"""
aafile = open('train_abnor3.list','w')
for a in aa:
    aafile.write(a)
random.shuffle(aa)
aafile.close()
aafile1 = open('train_abnor1.list','w')
for a1 in aa:
    aafile1.write(a1)
aafile1.close()
aafile2 = open('train_abnor2.list','w')
for a2 in aa:
    aafile2.write(a2)
aafile2.close()
"""
aafile = open('train_abnor3.list','w')
for a in aa:
    aafile.write(a)
random.shuffle(aa)
for a1 in aa:
    aafile.write(a1)
random.shuffle(aa)
for a2 in aa:
    aafile.write(a2)
aafile.close()

nnfile = open('train_normal.list','w')
for n in nn:
    nnfile.write(n)
nnfile.close()

合并两个生成的文件的内容，并将合并后的内容乱序

#合并两个文件的内容，并将合并后的内容乱序
import os
import random
nfile = open('train_normal.list','r')
nn = []
for line in nfile:
    nn.append(line)
nfile.close()
afile = open('train_abnor3.list','r')
for line in afile:
    nn.append(line)
#将列表中元素乱序
random.shuffle(nn)
rfile = open('train2.list','w')
for n1 in nn:
    rfile.write(n1)
rfile.close()