将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"one_answers" 3"

我想得到：

   1  3
1  0  2
3  0  2

是否有任何库会执行这样的操作，或者我需要在数组中手工做

Answer 1

pandas完成工作！

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)