Python 将多列数据写入 csv，并写入标题行

# 1. ROC 计算产生 fpr, tpr, threshold
fpr, tpr, thresholds = roc_curve(label, pred)

# 2. 指定要写入的文件名(csv无需自己创建)
file_name = 'fpr_tpr_threshold.csv'

# 3. 写入
csvfile = open(file_name, 'wt', encoding="UTF8")
writer = csv.writer(csvfile, delimiter=",")
# 标题行写入
header = ['fpr', 'tpr', 'thresholds']
# 数据写入
csvrow1 = []
csvrow2 = []
csvrow3 = []
csvrow1.extend(fpr)
csvrow2.extend(tpr)
csvrow3.extend(thresholds)
writer.writerow(header)
writer.writerows(zip(csvrow1, csvrow2, csvrow3))
csvfile.close()