2012-08-02 29 views
13

np.array该循环的结果有4383行和6列。我试过没有成功,使用matplotlib(pylab)中的pylab.imshow()来显示数组。目标是创建数组的图像,颜色梯度表示数组值的大小。 数组的每一行表示每天(4383天)湖温的深度变化。因此,目的是发现深度和时间上的湖泊温度差异。谢谢我怎样才能显示一个np.array与pylab.imshow()

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() 

回答

21

您可以使用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() 

enter image description here

你的情况,你可以检查的TempLake输出。