python绘制风场风羽图时2m/s以下的如何显示风杆而不是圆圈？

球求大人的球球

大乖乖 发表于 2024-11-2 09:01
图片供参考

谢谢，成功解决啦！

灭火器

提供一个不用修改 matplotlib 源码的函数版：

1. def custom_barbs(ax, *args, **kwargs):
   
2.     '''修改后的 Axes.barbs，使小于 half 的风速显示风杆而非圆圈。'''
   
3.     from matplotlib.path import Path
   
4.     from matplotlib.quiver import Barbs
   
5. 
   
6.     out_list = []
   
7. 
   
8.     def decorator(method):
   
9.         def wrapper(self, *args, **kwargs):
   
10.             mag = args[0]
    
11.             n_flags, n_barb, half_flag, empty_flag = method(self, *args, **kwargs)
    
12.             small_flag = empty_flag & (mag > 0)
    
13.             out_list.append(small_flag)
    
14.             empty_flag = mag == 0
    
15.             half_flag[small_flag] = True
    
16.             return n_flags, n_barb, half_flag, empty_flag
    
17. 
    
18.         return wrapper
    
19. 
    
20.     try:
    
21.         method = Barbs._find_tails
    
22.         Barbs._find_tails = decorator(method)
    
23.         barbs = ax.barbs(*args, **kwargs)
    
24.     finally:
    
25.         Barbs._find_tails = method
    
26. 
    
27.     small_flag = out_list[0]
    
28.     for i in np.nonzero(small_flag)[0]:
    
29.         verts = barbs._paths[i].vertices[:2]
    
30.         verts = np.vstack((verts, verts[0]))
    
31.         barbs._paths[i] = Path(verts)
    
32. 
    
33.     return barbs

复制代码

这样一来就能在一张图里同时实现带圆圈和不带圆圈的效果：

1. axes[0].barbs(lon, lat, u10, v10, regrid_shape=8, transform=crs)
   
2. custom_barbs(axes[1], lon, lat, u10, v10, regrid_shape=8, transform=crs)

复制代码

不过这个做法仅在 matplotlib 3.9 里测试过，更低的版本不确定能不能行。
无圆圈风向杆.ipynb (130.9 KB, 下载次数: 0)

2024-11-4 15:24 上传

点击文件名下载附件

球求大人的球球

灭火器 发表于 2024-11-4 15:24
提供一个不用修改 matplotlib 源码的函数版：
这样一来就能在一张图里同时实现带圆圈和不带圆圈的效果：
...

好的我去试试，谢谢！