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