在python中将一维数组转换为基于类的二维矩阵

Question

我使用多类分类器，因此为了在测试后对其进行评估，我需要将来自分类器（y_pred）的预测与真正的类值（y_test）进行比较。在

但我把它们都作为一维阵列，像这样：

y_test = [1, 1, 1, 2, 1, 4, 5, 3, ... etc ]
y_pred = [1, 1, 1, 2, 3, 2, 5, 0, ... etc ]

我总共有46节课。在

但是为了构建ROC曲线（如这里所示：http://scikit-learn.org/stable/auto_examples/plot_roc.html），我想我需要y_test和{}在一个具有二进制值的二维矩阵中，形状如下： number_of_test_cases x number_of_classes。在

其中每列表示一个类，1表示分类器在给定的测试样本行上识别了这个类。在

因此，鉴于我展示的上述几个值，我知道我需要y逯u测试，使其看起来像这样：

^{pr2}$

这就是我所理解的。。。我希望我是对的！在

有没有numpy函数从一维数组创建这样的矩阵？在

Answer 1

看看链接中的示例代码中引用的^{}函数。在