我正在使用Python 3.6。
我有一个元组列表,其中元组中的第一个元素是ID,第二个元素是预测(0或1)。我想创建一个标题为"PassengerId, Survived"
的CSV,示例行可能为"237, 1"
。
这是我的尝试:
import csv
with open("Predictions.csv", "wb") as f:
writer = csv.writer(f)
writer.writerow(["PassengerId", "Survived"])
for row in rows:
writer.writerow([row[0], row[1]])
不幸的是,该文件包含标题,但我没有尝试写入CSV。
知道我哪里出错了吗?
谢谢!