【Python代码】随机抽取文件名列表NameList中的Name作为训练集

#!/usr/bin/env python
#coding=utf-8

#随机抽取一部分图片作为测试集

import random

NameList=[]#存储所有图片名字
'''
NameListPath='NameList_v.txt' #图片名存储地址
SlicePath='Main visibal/test.txt' #抽取的元素的存储地址
SlicePath2='Main visibal/trainval.txt' #剩余的元素存储地址
'''
NameListPath='Main visibal/trainval.txt'#图片名存储地址
SlicePath='Main visibal/val.txt' #抽取的元素的存储地址
SlicePath2='Main visibal/train.txt' #剩余的元素存储地址

#slice_num=164 #抽取数量
slice_num=66 #抽取数量

file_object = open(NameListPath)
try:
    for line in file_object:
        '''line_new=line[ :-2] #截掉Windows下的\r\n
        line_new+='\n' #增加换行
        NameList.append(line_new)'''
        NameList.append(line)
finally:
    file_object.close( )

#随机抽取
slice = random.sample(NameList, slice_num)
#print slice

#存储随机抽取的元素
output1 = open(SlicePath, 'w')
try:
    output1.writelines(slice)
finally:
    output1.close()

#抽取后剩余
slice2=NameList
for name in slice:
    slice2.remove(name)
#print slice2

#存储随机抽取后剩余的元素
output2 = open(SlicePath2, 'w')
try:
    output2.writelines(slice2)
finally:
    output2.close()

print 'Done!'