如何使用pylab.imshow()显示np.array

Question

由此循环产生的np.array有4383行和6列.我试过没有成功使用matplotlib(pylab)中的pylab.imshow()来显示数组.目标是创建数组的图像,颜色渐变表示数组值的大小.阵列的每一行代表每天湖泊温度的深度变化(4383天).因此,目标是在深度和时间上找到湖泊温度的差异.谢谢

TempLake=np.zeros((N+1,Nlayers))
TempLake[0]=T0

Q=np.zeros(N+1)
Q[0]=0.0
for i in xrange(N):
    Q[i+1]=Qn(HSR[i],TD[i],FW[i],TempLake[i][0])
    TempLake[i+1]=main_loop(Z,z,Areat0,Areat1,TempLake[i],wind[i],Q[i],Q[i+1])


np.savetxt('test.out', TempLake, delimiter=',')
np.savetxt('test1.out', Q, delimiter=',')

pylab.imshow(TempLake)
pylab.show()

Answer 1

您可以使用imshow如下:

import pylab as plt
import numpy as np


Z=np.array(((1,2,3,4,5),(4,5,6,7,8),(7,8,9,10,11)))
im = plt.imshow(Z, cmap='hot')
plt.colorbar(im, orientation='horizontal')
plt.show()

在你的情况下,你可以检查输出TempLake.