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

Question

我有一个我已经破解的数据集

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放在同一数据帧中

Answer 1

添加reset_index()

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