python自定义数据集准备-写csv文件（标签-图片名）

import pandas as pd

#任意的多组列表
label_y=[]
image_name=[]
#print(image_name)
for i in range(30):
    image_name.append('{}.jpg'.format(i+1))
    if i<10:
        label_y.append(0)
    elif i>=10 and i<20:
        label_y.append(1)
    else:
        label_y.append(2)
#print(label_y)
#print(image_name)
#参考：http://t.csdn.cn/EwLFC、http://t.csdn.cn/GDAl1
#字典中的key值即为csv中列名
dataframe = pd.DataFrame({'label_y':label_y,'image_name':image_name})

#将DataFrame存储为csv,index表示是否显示行名，default=True
dataframe.to_csv("train_label.csv",index=False,sep=',')

 结果如下：