import matplotlib.pyplot as plt
import CSnative
def main() :
# The subscript starts with 0 in python
# User should specify the time, level, fieldname (var), and filename
Time=0
Level=0
var='T'
#filename = 'HU_getc180forweyium_c180.geosgcm_prog.20000414_2200z.nc4'
filename = 'TEST7.geosgcm_prog.20000415_0000z.nc4'
lons,lats,data = CSnative.get_csdata(filename, var, Time, Level);
# color map plot
plt.figure()
plt.tripcolor(lons,lats,data.T,vmin=data.min(), vmax=data.max())
plt.axis([lons.min(), lons.max(), lats.min(), lats.max()])
plt.colorbar()
plt.draw()
plt.show(block=False)
# plt.savefig('T1.png')
# contour plot
plt.figure()
plt.tricontour(lons,lats,data.T, 20, linewidths=0.5,cmap=plt.cm.rainbow )
plt.axis([lons.min(), lons.max(), lats.min(), lats.max()])
plt.colorbar()
plt.draw()
plt.show()
if __name__ == '__main__' :
main()