Python oneliner提供所需的混乱/应急矩阵

Question

我想写一个一行程序来计算一个混淆/列联矩阵M（任何一个维度都等于类的个数的平方矩阵），它计算两个长度为n的向量：Ytrue和y预测的情况。显然，使用python和numpy时，以下操作不起作用：

error = N.array([error[x,y]+1 for x, y in zip(Ytrue,Ypredicted)]).reshape((n,n))

有什么提示可以创建一个线性矩阵混淆计算器吗？在

Answer 1

如果NumPy更新或等于1.6，并且Ytrue和Ypred是NumPy数组，那么这段代码可以工作

np.bincount(n * (Ytrue - 1) + (Ypred -1), minlength=n*n).reshape(n, n)