write_csv

在windows下，将数据写入CSV文件，代码如下:

import csv
def write_csv(data, filename):
    with open(filename, 'w', newline='') as csvfile:
        writer = csv.writer(csvfile)
        writer.writerow(['ImageId','Label'])
        for d in data:
            writer.writerow([str(i), str(d)])

需要注意几点：

• 如果open中没有newline参数，输出文件中会增加多于的空行（类似于print的输出，如果输出完不换行，应该print(" ", end='')）
• wirterow函数可以使得一个list中所有数据当做一行输入，而writerows会根据list中的,进行换行处理
• 数据用转换成str，其他类型行不通