python制作labels.csv文件的办法

学了python用起来

#制作之前的标签文件，训练集400张，四类0 1 2 3，各100张
ls = []
for j in range(4):
    for i in range(100*j+1, 100*(j+1)+1):
        ls.append([str(i), str(j)])
f = open('labels.csv', 'w')
for item in ls:
    f.write(','.join(item) + '\n')

f.close()

注意生成二维列表的时候内层列表的每一个元素必须是字符串，因此要str()一下