Python文件处理：数据标签修改

今天在做眼底图像识别的时候，发现0级别和1，2，3，4级别的数量严重不等，所以想到了设计一个分级的卷积神经网络，先进行二分类，即0为一类，2，3，4为一类，这也就涉及到了训练集标签的修改了，所以用python写了一个小脚本。

train = open('trainLabels.txt','r')
train_new = open('trainLabels_new.txt','w+')
line = train.readline()
while line:
    line = list(line)
    label = line[len(line)-2