我有一个矩阵，我想将其转换为csv。所以我写：
np.savetxt('titanic1.csv', Sol, fmt='%.2f', delimiter=",",header="PassengerId,Survived")
但是我得到的标题是“#PassengerId”而不是我想要的“PassengerId”。我该如何解决这个问题？
从标题中删除#
数据挖掘
数据集
CSV
矩阵
2022-03-06 10:05:36
2个回答
numpy的文档说您可以添加参数comments=''，但警告说 np.loadtxt() 可能不适用于生成的文件。
您可以使用以下代码替换您的标题：
for row in fileinput.input('titanic1.csv', inplace=True):
if fileinput.isfirstline():
print('PassengerID,Survived')
else:
print(row)