协调底图箭袋和 matplotlib 箭头

Question

我正在尝试比较 matplotlib 中网格化模型输出位置（通过底图地图上的箭袋）和分散站（通过 matplotlib 箭头）之间的风矢量。两者的位置都在 lat/lon，但风矢量在 m/s。

合并后，我希望颜色和长度随大小而变化，并且对于箭袋和箭头数据，两种质量都以相同的方式缩放。我在下面给出了一个示例，其中箭袋图看起来不错并且按绝对长度（英寸）缩放。我不知道如何做才能使 arrow() 匹配。在示例中，我将其除以 SCALE，以了解我希望最终图像看起来像什么。

import numpy as np
import matplotlib.pylab as plt
from mpl_toolkits.basemap import Basemap

X, Y = np.meshgrid(np.arange(-123,-121,0.3),np.arange(37,39,0.3))
U = np.cos(X+123)*12
V = np.sin(Y-37)*12
mag = np.hypot(U,V)

fig,ax=plt.subplots(1)
m=Basemap(projection ='cyl',resolution='f',llcrnrlat=37,llcrnrlon=-123,
          urcrnrlat=39,urcrnrlon=-121,ax=ax)

quiv = m.quiver(X,Y,U,V,mag,zorder=2,latlon=True,scale=30,scale_units='inches')

# Scattered points won't be on the grid
x0=X[2,2] - 0.025
y0=Y[2,2]

u0=U[2,2]
v0=V[2,2] + 0.5 
SCALE = 72.
plt.arrow(x0,y0,u0/SCALE,v0/SCALE)

plt.show()

Answer 1

matplotlib 文档（在我看来）不是很清楚，但是 quiver 接受所有 X、Y、U 和 V 的一维数组，不需要均匀间隔。底图文档弄错了，或者至少更加不清楚。因此，只要将分散的站数据形成一维数组，就可以了。

我通过用这个替换你的散点部分向你的情节添加了一些随机箭头（如果你使用相同的种子你应该得到相同的箭头）：

# Make scattered locations
np.random.seed(33)
x0 = np.random.rand(5)*2.0 - 123
y0 = np.random.rand(5)*2.0 + 37

# Make some velocities
u0 = np.random.randn(5)*3 + 10
v0 = np.random.randn(5)*3 + 10

q2 = m.quiver(x0, y0, u0, v0, latlon=True, scale=30, scale_units='inches')

这是我得到的图（我默认使用 YlGnBu_r 颜色图）。

请注意，如果您开始使用圆柱投影以外的任何投影（并且如果您的 U 和 V 以东西向和南北向表示），您将需要旋转矢量以匹配投影使用 rotate_vector 方法。

Answer 2

您需要通过x,y = m(lon, lat)将地图坐标转换为笛卡尔坐标，此后plt.quiver(x,y, u, v)或m.quiver(x,y,u,v)将完成相同的工作。

协调底图箭袋和 matplotlib 箭头

Coordinating basemap quiver and matplotlib arrow

python

numpy

matplotlib

matplotlib-basemap