quiver画风速的参考风速问题

zhanghm2018

这个代码可以绘制风速，绘制的风速受到quiver函数的scale参数的影响会变化长度。因此需要在图中增加一个参考值（也即红色的风速代表10m/s） 如何让这个标志出现在色卡的上方？

zhanghm2018

这是上述代码的绘制结果

zhanghm2018

下图为想要实现的结果

灭火器

把 QuiverKey 的 _set_transform 私有方法覆盖掉。

1. import matplotlib.pyplot as plt
   
2. from matplotlib.cm import ScalarMappable
   
3. 
   
4. fig, ax = plt.subplots()
   
5. mappable = ScalarMappable()
   
6. cbar = fig.colorbar(mappable, ax=ax)
   
7. 
   
8. Q = ax.quiver(0.5, 0.5, 1, 1, scale=20)
   
9. qk = ax.quiverkey(Q, X=0.5, Y=1.05, U=2, label='2 m/s')
   
10. qk._set_transform = lambda: qk.set_transform(cbar.ax.transAxes)
    
11. 
    
12. plt.show()

复制代码

Nozigaka46

感谢，学习一下

zhanghm2018

灭火器 发表于 2024-5-18 22:11
把 QuiverKey 的 _set_transform 私有方法覆盖掉。

妙啊 我就是出不了Q的那个ax。 学到了

灭火器

zhanghm2018 发表于 2024-5-19 12:04
妙啊 我就是出不了Q的那个ax。 学到了

直觉来说让 QuiverKey 使用 colorbar 的 Axes 的坐标系就行，就是指定参数 transform=cbar.ax.transAxes 就行。然后很简单就可以把参考风速定位到 colorbar 的正上方。
但神秘的是 QuiverKey 没有 transform 参数，而是用的 coordinates 参数，所以只能用这种“篡改”底层的方法。

zhanghm2018

灭火器 发表于 2024-5-19 12:24
直觉来说让 QuiverKey 使用 colorbar 的 Axes 的坐标系就行，就是指定参数 transform=cbar.ax.transAxes  ...

1. import matplotlib.pyplot as plt
   
2. from matplotlib.cm import ScalarMappable
   
3. 
   
4. fig, ax = plt.subplots()
   
5. mappable = ScalarMappable()
   
6. cbar = fig.colorbar(mappable, ax=ax)
   
7. 
   
8. 
   
9. 
   
10. Q = ax.quiver(0.5, 0.5, 5, 0, scale=20)
    
11. qk = ax.quiverkey(Q, X=1.1, Y=1.02, U=2, label='2 m/s')
    
12. # qk._set_transform = lambda: qk.set_transform(cbar.ax.transAxes)
    
13. plt.title('aaaaa')
    
14. plt.show()

复制代码

zhanghm2018

灭火器 发表于 2024-5-19 12:24
直觉来说让 QuiverKey 使用 colorbar 的 Axes 的坐标系就行，就是指定参数 transform=cbar.ax.transAxes  ...

我尝试了一下 直接在QuiverKey上就可以改。单个图是可以的。

灭火器

因为 quiverkey 方法默认就是使用 quiver 所在的 Axes 的坐标系，所以你指定 X=1.1 也行。
不过当 Axes 或 colorbar 的大小、位置发生变化，X=1.1 的位置就不一定对应 colorbar 中间正上方了。
所以我用的是 colorbar 的坐标系，指定 X=0.5，即中间一半。