z是一个二维矩阵,使用imshow画出来的图相当于用颜色看每一行每一列的值
import numpy as np
import matplotlib.pyplot as plt
#生成一个随机的z
z=np.reshape(np.random.randint(-10,10,12),[3,4])
print(z)
#如果cmap是coolwarm_r则颜色是反过来的
plt.imshow(z,cmap='coolwarm',origin='lower')
plt.colorbar()
'''如果需要为colorbar添加label,否则直接用上面那个就行
cb=plt.colorbar()
cb.set_label('z label')'''
plt.show()
输出:
[[-7 9 5 1]
[-8 2 3 0]
[ 2 -9 2 8]]
可以看到是默认按行列的index绘图,图像的形状也是和直接输出矩阵是一样的,只是值换成了颜色。如果想要把纵轴的0放到下面,则需要将imshow那一行改为
plt.imshow(z,cmap='coolwarm',origin='lower')
然后如果需要进一步调整坐标轴以对应特定的x和y,则需要通过plt.xticks()对坐标轴刻度和值进行改变