python - 如何打印二维矩阵并添加标签，同时保持所有内容居中 (Python)

Question

我有一个混淆矩阵和一组标签，我想在打印时将其添加到每一行和每一列

这是我的代码:

def PrintConfusionMatrix(matrix,labels):
    for i in range(len(labels)):
        print(labels[i],end = "  ")#print labels on the first line

    for i in range(len(mc)):
        print("\n")
        print(labels[i], end = "  ")
        for j in range(len(mc)):
            print(matrix[i][j], end = "  ")

输入如下:

labels = ['Iris-setosa', 'Iris-versicolor', 'Iris-virginica']
matrix = [[19,  0,  0],
         [ 0, 19,  0],
         [ 0,  2, 20]]

它当前打印此:

Iris-setosa  Iris-versicolor  Iris-virginica  

Iris-setosa  19  0  0  

Iris-versicolor  0  19  0  

Iris-virginica  0  2  20

但我希望它以这种方式打印

             Iris-setosa  Iris-versicolor  Iris-virginica  

Iris-setosa       19              0               0  

Iris-versicolor    0             19               0  

Iris-virginica     0              2              20

有人可以帮我解决这个问题吗？如果它可以扩展到更大的二维矩阵，那就太好了

谢谢

Answer 1

如果您不介意使用外部库，一个简单的方法是使用 pandas:

import pandas as pd

df = pd.DataFrame(matrix, columns=labels, index=labels)

print(df)

                 Iris-setosa  Iris-versicolor  Iris-virginica
Iris-setosa               19                0               0
Iris-versicolor            0               19               0
Iris-virginica             0                2              20