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

Question

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有shape：(3000,)

我想将该列重新添加到X_test标题中job_performance，然后最后保存为 CSV。有人能帮忙吗？

标签： pythonpandasscikit-learn

Answer 1

然后像这样保存到csv

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