python在csv文件的头文件中添加列表和字符串

Question

a=[]

    for i in range(9216):

        a.append("pixel"+str(i))

    with open('train41.csv', 'w') as csvfile:

        writer = csv.DictWriter(csvfile, fieldnames=["label",a])
        writer.writeheader()

这不起作用

我想在我的csv文件头

中添加一个字符串标签和列表a

Answer 1

fieldnames必须是字符串列表：

writer = csv.DictWriter(csvfile, fieldnames=["label"] + a)

您使用的列表包含一个字符串和另一个字符串列表：

writer = csv.DictWriter(csvfile, fieldnames=["label",a])