我有一个形状为(3,3)
的数组A
.有没有办法在3x3大小的正方形上表示数组元素?通常,我想在大小为nxn
的正方形上表示nxn
个数组?已附加预期输出.
import numpy as np
A=np.array([[10,20,30],[40,50,60],[70,80,90]])
预期输出为:
我有一个形状为(3,3)
的数组A
.有没有办法在3x3大小的正方形上表示数组元素?通常,我想在大小为nxn
的正方形上表示nxn
个数组?已附加预期输出.
import numpy as np
A=np.array([[10,20,30],[40,50,60],[70,80,90]])
预期输出为:
您可以使用具有良好API的seaborn.heatmap
:
import numpy as np
import seaborn as sns
from matplotlib.colors import ListedColormap
A = np.array([[10, 20, 30], [40, 50, 60], [70, 80, 90]])
ax = sns.heatmap(A,
annot=True, square=True, cbar=False,
xticklabels=False, yticklabels=False,
cmap=ListedColormap(['white']),
linecolor='k', lw=2,
annot_kws={'size': 30}
)
ax.figure.savefig('img.png')
输出: