R语言predict函数输出格式调整:替换索引与结果列命名
解决方案
你可以通过以下两种简洁的方式实现需求,核心是将predict()的结果与原df_smooth合并,并为预测值列指定有意义的名称:
方法一:直接在原数据框中添加预测列
直接在df_smooth中新增一列存储预测值,一步完成列添加与命名:
# 为预测结果指定列名(比如对应模型的因变量含义,这里用predicted_win_pct) df_smooth$predicted_win_pct <- predict(model, df_smooth) # 查看处理后的结果 head(df_smooth)
执行后,df_smooth会包含两列:spread_favorite(原自变量列,作为普通列显示而非索引)和predicted_win_pct(重命名后的预测结果列)。
方法二:合并数据框生成新结果集
如果不想修改原df_smooth,可以先将预测结果转为带列名的数据框,再与原数据框合并:
# 生成带指定列名的预测结果数据框 prediction_df <- data.frame(predicted_win_pct = predict(model, df_smooth)) # 合并原数据框与预测结果 result_df <- cbind(df_smooth, prediction_df) # 查看合并后的结果 head(result_df)
这种方式会生成一个新的数据框result_df,同样包含spread_favorite和命名后的预测列,原df_smooth保持不变。
内容的提问来源于stack exchange,提问作者Abartel
相关产品推荐
相关产品推荐

