求助:python中用plt绘制图形时,如何限定图形尺寸(要使横纵坐标不等比例)?

需要将矩阵可视化,生成一个正方形的热力图,但原矩阵不是方阵,导致输出的图形始终是个矩形。

尝试了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()

图片说明

查看全部
weixin_45857220
Cody Pong
2020/04/08 11:01
  • python
  • 点赞
  • 收藏
  • 回答
    私信

2个回复