如何摆脱行号，to_csv？

Question

在预测分类问题的目标值后，尝试在 .csv 文件中获取预测值以及特定数据实例的 id，但获取了不必要的行号。

x_train,x_test,y_train,y_test = train_test_split(x,y, random_state=101,strtify =y)

cls = DecisionTreeClassifier()
cls.fit(x_train,y_train)

pred = cls.predict(x_test)
prediction = pd.DataFrame({'id':x_test['id'],'target':pred})
prediction.to_csv("path/prediction.csv")

x_test.head() image

to.csv() image

Answer 1

你 index=False , doc 指定这个参数

prediction.to_csv("path/prediction.csv",index=False)