如何在 matplotlib.pyplot.quiver 图中获取箭头方向（即 U 和 V）？

Question

我想知道在绘制matplotlib.pyplot.quiver 之后，是否有任何方法可以使用Axes 对象来获取箭头方向（即U 和V）。例如：

fig, ax = plt.subplots()
plt.quiver(X=..., Y=..., U=..., V=...)
ax.collections[0].something_that_returns_UV?

我知道ax.collections[0].get_offsets() 会返回X 和Y 值，但我对U 和V 值感兴趣。

用例：自动测试绘图 (example)。

Answer 1

ax.quiverkey 创建的集合是matplotlib.quiver.Quiver 类型。可以直接获取X、Y、U、V为ax.collections[0].X等。

import matplotlib.pyplot as plt
import numpy as np

X, Y = np.meshgrid(np.arange(0, 2 * np.pi, .2), np.arange(0, 2 * np.pi, .2))
U = np.cos(X)
V = np.sin(Y)
fig1, ax = plt.subplots()
Q = ax.quiver(X, Y, U, V)

print('X:', ax.collections[0].X)
print('Y:', ax.collections[0].Y)
print('U:', ax.collections[0].U)
print('V:', ax.collections[0].V)