需要将矩阵可视化,生成一个正方形的热力图,但原矩阵不是方阵,导致输出的图形始终是个矩形。
尝试了plt.figure(figsize=(6, 6));但只能让输出的图片是正方形(绘图区还是一个矩形,结果见后图)。
恳请大神赐教,不胜感激!
import numpy as np
import matplotlib.pyplot as plt
#data为8x4的矩阵
data=np.array([[1,3,5,2],[2,6,4,9],[4,7,0,3],[8,5,1,7],[1,3,6,2],[2,8,4,0],[3,7,9,1],[8,5,1,7]])
plt.figure(figsize=(6, 6))
cmap=plt.cm.inferno#设置热力图的颜色样式
img=plt.imshow(data,cmap=cmap)
plt.colorbar()