python画风垂直切变填色图

huyy

1. # -*- coding: utf-8 -*-
   
2. from mpl_toolkits.basemap import Basemap
   
3. import numpy as np
   
4. #from matplotlib import cm
   
5. import matplotlib.pyplot as plt
   
6. from matplotlib import colors
   
7. from netCDF4 import Dataset
   
8. import scipy.ndimage
   
9. 
   
10. xmin=40
    
11. xm=57
    
12. xmax=73
    
13. ymin=24
    
14. ymax=37
    
15. ny=13
    
16. nx=33
    
17. st=0
    
18. mt=48
    
19. et=97
    
20. nt=96
    
21. #读取数据计算垂直风切变
    
22. a=Dataset("D:\\cygwin\\cygdrive\\dachuang\\uv2.5.nc","r")
    
23. z1=0
    
24. z2=1
    
25. uwnd=a.variables['u']
    
26. uwnd1=uwnd[:,z1,:,:]
    
27. uwnd2=uwnd[:,z2,:,:]
    
28. vwnd=a.variables['v']
    
29. vwnd1=vwnd[:,z1,:,:]
    
30. vwnd2=vwnd[:,z2,:,:]
    
31. 
    
32. #计算垂直风切变
    
33. vsheer=np.sqrt((uwnd2-uwnd1)*(uwnd2-uwnd1)+(vwnd2-vwnd1)*(vwnd2-vwnd1))
    
34. 
    
35. vsheer1=vsheer[st:mt,:,:]
    
36. vsheer1=np.mean(vsheer1,axis=0)
    
37. vsheer2=vsheer[mt:et,:,:]
    
38. vsheer2=np.mean(vsheer2,axis=0)
    
39. vsheer3=(vsheer2-vsheer1)
    
40. vsheer3[vsheer3>=3]=0
    
41. vsheer3[vsheer3<=-2]=0
    
42. vsheer3=scipy.ndimage.gaussian_filter(vsheer3,1)
    
43. 
    
44. bidate=a.variables['time']
    
45. blat=a.variables['latitude'][:]
    
46. blon=a.variables['longitude'][:]
    
47. blons,blats=np.meshgrid(blon,blat)
    
48. 
    
49. ax=plt.subplot(1,1,1)
    
50. # setup mercator map projection.
    
51. m = Basemap(llcrnrlon=100.,llcrnrlat=0.,urcrnrlon=180.,urcrnrlat=30.,\
    
52.             rsphere=(6378137.00,6356752.3142),\
    
53.             resolution='l',projection='merc',\
    
54.             lat_0=15.,lon_0=140.,lat_ts=20.)
    
55. m.drawcoastlines(linewidth=0.3)
    
56. #m.fillcontinents(color='linen')
    
57. 
    
58. blonm,blatm=m(blons,blats)
    
59. 
    
60. #调色
    
61. #'#2F4F4F','#008080','#008B8B','#40E0D0','#00FFFF','#7FFFD4','#F0FFFF',
    
62. #'#F0FFFF','#FFFF00','#FFA500','#FF8C00','#FF4500','#FF0000'
    
63. #'#9400D3','#0000FF','#4169E1','#00BFFF',
    
64. cdict=['#008080','#40E0D0','#00FFFF','#FFFFFF',\
    
65. '#F0FFFF','#FFFF00','#FFA500','#FF8C00']
    
66. my_cmap=colors.ListedColormap(cdict,'indexed')
    
67. #cm.register_cmap(name ='dbzcmap',data = cdict,lut = 128)
    
68. # 注意：cpool中有15个颜色，但是只使用了10个，其余的会被忽略
    
69. # 具体可查阅matplotlib官方文档中对ListedColormap的说明
    
70. 
    
71. #lev=np.linspace(np.min(sst3),np.max(sst3),50)
    
72. #bounds=np.linspace(-3.2,2.4, 8, endpoint=True)
    
73. norm=colors.Normalize(vmin=-2.0,vmax=3)
    
74. cf=m.contourf(blonm,blatm,vsheer3,cmap=plt.cm.RdBu_r,norm=norm)
    
75. cb=m.colorbar(cf,"bottom",size="10%",pad=0.3,alpha=1,norm=norm)
    
76. #,ticks=bounds,boundaries=bounds)
    
77. #cb.set_ticks(np.arange(-1.2,2.8,0.4))
    
78. #cb.set_ticklabels(np.arange(-1.2,2.8,0.4))
    
79. #cbar = pyplot.colorbar(pp, orientation='vertical', ticks=np.arange(cbar_min, cbar_max+cbar_step, cbar_step), format=cbar_num_format)
    
80. m.drawparallels(np.arange(0,40,10),labels=[1,0,0,0],fontsize=10,linewidth=0.0,)
    
81. # draw meridians
    
82. m.drawmeridians(np.arange(100,220,20),labels=[0,0,0,1],fontsize=10,linewidth=0.0)
    
83. ax.set_title('Vms',loc='left')
    
84. plt.show()

复制代码

213de88

帅哥要能把图分享了就更好了

abodoco

42.vsheer3=scipy.ndimage.gaussian_filter(vsheer3,1) 请问这一句是在做什么处理？为什么需要滤波？

famousky

谢谢分享，还是第一个

许市

楼主最好给出图片示例，方便大家学习

linlinouc

请问楼主资料是什么资料呢？有处理过么？

huyy

linlinouc 发表于 2018-8-26 19:51
请问楼主资料是什么资料呢？有处理过么？

ECMWF风场资料，未处理

huyy

许市 发表于 2018-8-24 08:43
楼主最好给出图片示例，方便大家学习

好的，以后注意~

213de88

帅哥要能把图分享了就更好了

嘴角上扬

出来的图长啥样呢