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

标签 python pandas scikit-learn

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

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_pred形状:(3000,)

我想将该列添加回 X_test,标题为 job_performance,然后最后另存为 CSV。有人可以帮忙吗?

最佳答案

然后像这样保存到csv

X_test['job_perf'] = y_pred
X_test.to_csv('filename.csv')

关于python - 如何在 Pandas DataFrame 中为我的预测结果添加一列,然后另存为 CSV?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56761135/

相关文章:

python - 名称错误 : name 'fish' is not defined

python - kde 情节 : UserWarning: Dataset has 0 variance; skipping density estimate

python - scikit-learn FeatureUnion 网格搜索特征子集

machine-learning - 为什么 scikit-learn SVM 分类器交叉验证如此慢?

scikit-learn - sklearn中分类数据的递归特征消除?

python - 需要从 Python 将指针传递给 C 共享库函数

python - pexpect 和 ssh 连接到 ipv6 主机

Python @property 与 @property.getter

python - Pandas 空数据帧由 isin 函数生成,如果 ID 存在于仅包含 ID 的数据帧中,则该函数会保留具有 ID 的对象

python - Pandas cut 方法为值生成了错误的类别