python快速绘图

Hua_Zhang

python快速绘图，使用python简单绘制Anomaly SST的空间分布(python的nc文件读取，basemap的调用)

1. import numpy as np
   
2. import matplotlib.pyplot as plt
   
3. import netCDF4 as nc
   
4. from mpl_toolkits.basemap import Basemap as mp
   
5. 
   
6. indir='D:\\python_plot\\'
   
7. obj = nc.Dataset(indir+'1979-2017.nc','r')
   
8. for i in obj.variables.keys():
   
9.     print(i)  
   
10. print('---------------------------------------')
    
11. print(obj)
    
12. lat = obj.variables['latitude'][:]
    
13. lon = obj.variables['longitude'][:]
    
14. time = obj.variables['time'][344:]
    
15. sst = obj.variables['sst'][:,:,:]
    
16.                
    
17. sst_1 = sst[344:,:,:]
    
18. sst_aj = np.zeros((len(time),len(lat),len(lon)))
    
19. for i in range(len(lat)):
    
20.     for j in range(len(lon)):
    
21.         for k in range(len(time)):
    
22.             if sst_1[k,i,j] == -1000.0:
    
23.                 sst_aj[k,i,j] = np.nan
    
24.             else:
    
25.                 sst_aj[k,i,j] = sst_1[k,i,j]
    
26. sst_mean = np.mean(sst_aj,0)
    
27. sst_g = sst_1[119,:,:] - sst_mean
    
28. 
    
29. plt.figure(1, figsize=(10, 8))
    
30. m = mp(llcrnrlon=-180, llcrnrlat=-85., urcrnrlon=180., urcrnrlat=85., resolution='c', projection='mill')
    
31. m.drawcoastlines
    
32. m.drawparallels(np.arange(-90, 90, 30),labels=[1, 0, 0, 0])
    
33. m.drawmeridians(np.arange(m.lonmin, m.lonmax+30, 60), labels=[0, 0, 0, 1])
    
34. cmap = plt.cm.get_cmap('RdYlBu')
    
35. levels = np.arange(-5, 5, 0.7)
    
36. xx, yy =np.meshgrid(lon, lat)
    
37. xx, yy = m(xx, yy)
    
38. m.contourf(xx, yy, sst_g, cmap=cmap,levels=levels)
    
39. m.colorbar(label='sst', location='bottom', pad=0.3)
    
40. plt.title(' Anomaly SST ',loc='left', fontsize=18)
    
41. # =============================================================================
    
42. # plt.show()
    
43. # =============================================================================
    
44. plt.savefig('Anomaly_SST.png',dpi=520)
    
45. print('ok!!')

复制代码

rceclx

不错，lz用Python绘图厉害啊！

kingjames

楼主，数据能分享一份吗？

EddyLiu

basemap安装不上啊

tancr

EddyLiu 发表于 2020-3-1 18:19
basemap安装不上啊

改用cartopy

vicky9898

太感谢了 对我很有启发！！