python - 如何在极坐标中制作箭袋图

Question

如何在极坐标中制作箭袋图？我有关于 r 和 theta 的数据。我试过了：

import numpy as np

radii = np.linspace(0.5,1,10)
thetas = np.linspace(0,2*np.pi,20)
theta, r = np.meshgrid(thetas, radii)

f = plt.figure()
ax = f.add_subplot(111, polar=True)
ax.quiver(theta, r, dr, dt)

其中 dr 和 dt 是 r 和 theta 方向的数据向量。

Answer 1

看起来 quiver 不会为您进行转换。您需要手动进行 (r,t) -> (x,y) 转换：

radii = np.linspace(0.5,1,10)
thetas = np.linspace(0,2*np.pi,20)
theta, r = np.meshgrid(thetas, radii)

dr = 1
dt = 1

f = plt.figure()
ax = f.add_subplot(111, polar=True)
ax.quiver(theta, r, dr * cos(theta) - dt * sin (theta), dr * sin(theta) + dt * cos(theta))