开发者社区> 问答> 正文

如何将测试数据帧放在预测数据帧的同一数据帧中?

我有数据集 X_train, X_test, y_train, y_test = train_test_split(X,y, random_state =34) 我已经构建了我的模式并获得了预测 mdl.fit(X_train, y_train) y_pred = mdl.predict(X_test) 现在我想单独看一下预测值和y_test值 我这样做了 pd.DataFrame(y_pred).join(y_test) 但结果是NaN大多数值

             0  AdmAfter12Mths

0 1.113033e+00 NaN 1 1.794673e+00 NaN 2 -1.279929e-01 NaN 3 8.467617e-01 NaN 4 1.315781e+00 NaN 5 5.421329e-01 NaN 6 1.203635e-01 NaN 7 9.216141e+00 NaN 8 1.248840e+01 NaN 9 1.027268e+00 7.0 10 3.767157e-02 NaN 我尝试了另一种方式

pd.concat([y_test, pd.DataFrame(y_pred)], axis=1) 相同的结果

            0  AdmAfter12Mths

0 1.113033 NaN 1 1.794673 NaN 2 -0.127993 NaN 3 0.846762 NaN 4 1.315781 NaN 5 0.542133 NaN 6 0.120363 NaN 7 9.216141 NaN 8 12.488396 NaN 9 1.027268 7.0 10 0.037672 NaN 我意识到数据帧索引可能是问题所在,但, 我怎样才能把两者y_pred并y_test在同一个数据帧

展开
收起
游客6qcs5bpxssri2 2019-09-07 21:42:27 582 0
1 条回答
写回答
取消 提交回答
  • 添加 reset_index()

    pd.concat([y_test.reset_index(drop=True), pd.DataFrame(y_pred)], axis=1)

    2019-09-07 21:43:02
    赞同 展开评论 打赏
问答分类:
问答标签:
问答地址:
问答排行榜
最热
最新

相关电子书

更多
移动互联网测试到质量的转变 立即下载
给ITer的技术实战进阶课-阿里CIO学院独家教材(四) 立即下载
F2etest — 多浏览器兼容性测试整体解决方案 立即下载