首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >如何在Pandas DataFrame中为我的预测结果添加一列,然后保存为CSV?

如何在Pandas DataFrame中为我的预测结果添加一列,然后保存为CSV?
EN

Stack Overflow用户
提问于 2019-06-26 03:54:06
回答 1查看 46关注 0票数 2

如果重要的话,我会使用sklearn,但我有:

代码语言:javascript
复制
regressor = RandomForestRegressor(
    n_estimators=10, random_state=0, verbose=1, n_jobs=-1)
regressor.fit(X_train, np.ravel(y_train))
y_pred = regressor.predict(X_test)

我的y_predshape(3000,)

我想将该列添加回X_test,标头为job_performance,最后保存为CSV。有人能帮上忙吗?

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2019-06-26 04:06:27

然后保存为csv,如下所示

代码语言:javascript
复制
X_test['job_perf'] = y_pred
X_test.to_csv('filename.csv')
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/56761135

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档