Matplotlib 矩阵相关图

可以使用 Matplotlib 创建相关图。Matplotlib 是 Python 最常用的绘图库。它可以包含在可用于 Python 的所有图形工具包中。

矩阵定义

首先,我们定义一个 2×2 矩阵和一个名为 groups 的列表。矩阵在括号内定义。

m = [
[1,0,2,0,0],
[1,1,1,2,0],
[0,4,1,0,0],
[0,4,4,1,2],
[1,3,0,0,1],
]

groups = ['Blues','Jazz','Rock','House','Dance']

图形

matrixCorrelation

矩阵相关图

下面的代码使用 Matplotlib 生成 Matrix 相关图。

import matplotlib.pyplot as plt
import numpy as np

m = [
[1,0,2,0,0],
[1,1,1,2,0],
[0,4,1,0,0],
[0,4,4,1,2],
[1,3,0,0,1],
]

plt.matshow(m)

groups = ['Blues','Jazz','Rock','House','Dance']

x_pos = np.arange(len(groups))
plt.xticks(x_pos,groups)

y_pos = np.arange(len(groups))
plt.yticks(y_pos,groups)

plt.show()

最初我们定义矩阵 m 和列表 groups

我们将长度设置为等于 gropus 的长度。在 x 轴和 y 轴上,我们设置了 groups 里面的名称。