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']
图形
矩阵相关图
下面的代码使用 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 里面的名称。