Randomforest分类器输出

Question

在我的型号打印出输出后，看起来像这样的东西，

id        val
234       0.99
235       0.3
236       0
237       0.92

我希望Val列是二进制的，因此阈值也是如此。

test[test["val"] < 0.5] = 0
test[test["val"] >= 0.5] = 1

阈值后，由于某种原因，我的输出看起来像这样，

id        val
1         1
0         0
0         0
1         1

我该如何确保ID列不会更改？

Answer 1

您提供的条件仅给出应更改的行。使用loc，并提供应更改的列。

test.loc[test["val"] < 0.5, "val"] = 0
test.loc[test["val"] >= 0.5, "val"] = 1

输出：

    id  val
0  234  1.0
1  235  0.0
2  236  0.0
3  237  1.0