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

Question

此循环产生的np.array结果有 4383 行和 6 列。我尝试使用pylab.imshow()from matplotlib(pylab)来显示数组但没有成功。

目标是创建数组的图像，其中颜色渐变表示数组值的大小。

数组的每一行代表每天（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.