小技巧（4）：使用Python将txt中的某两列数据写入csv文件中，制作图像分类标签

上一篇：小技巧（3）：简简单单地使用python读写txt文件

TSRD数据集的标签下载下来是这样的，做图像分类的话，我只需要第一列和最后一列。

import csv
	
	csvfile_path = 'train_data.csv'
	txtfile_path = "TsignRecgTrain4170Annotation.txt"
	example = []
	
	for line in open(txtfile_path):
	    del example[:]
	    a = line.split(';')
	    b1 = a[0]   # 这是选取需要读取的位数
	    b2 = a[7]
	    b2 = int(b2)
	    print(b1)
	    print(b2)
	    example.append(b1)
	    example.append(b2)
	    # 将列表中个数据写入CSV文件
	    with open(csvfile_path,'a',newline='') as csvfile:
	        writer = csv.writer(csvfile)
	        writer.writerow(example)

生成的train_data.csv文件是这样的

下一篇：小技巧（5）：将TT100K数据集转成VOC格式，并且用Python脚本选出45类超过100张的图片和XML