使用标签将Numpy混淆矩阵保存在人类可读文本中

Question

我对numpy并不是非常熟悉，但我正在使用它来计算一个混淆矩阵，除了它不会在x和y轴上打印标签之外，它很棒。由于我的数据标签并不总是一样的，所以回去检查它们的顺序是很痛苦的。现在我正在使用：

现在我正在使用：

true_val = [int(i) for i in y]
predict_val = [int(i) for i in y_pred]
confusion = confusion_matrix(true_val, predict_val)
np.savetxt('confusion_matrix.txt', confusion, delimiter=',')

现在我得到一个这样的矩阵：

[[0 2]
[0 2]]

但是说我的标签名为“1”和“3”

我想得到：

   1  3
1  0  2
3  0  2

是否存在可以执行此类操作的库，或者是否需要在数组中手动​​执行此操作

Answer 1

熊猫做到了！

import pandas as pd

d = {'1' : pd.Series([0,0], index=[1,3]),
     '3' : pd.Series([2,2], index=[1,3])}
df = pd.DataFrame(d)
print(df)