如何使用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的输出。