你如何用极坐标图 pcolormesh ？

Question

我想使用极坐标投影给出的漂亮圆，但我有笛卡尔数据。 我现在能做的最好的是以下，但它有一个缺口。

fig, ax = plt.subplots(subplot_kw={'projection': 'polar'})

xs, ys = np.meshgrid(np.linspace(-1,1,50),np.linspace(-1,1,50),)
parab = xs**2+ys**2

rs = np.sqrt(xs**2+ys**2)
ths = np.arctan2(ys, xs)

ax.pcolormesh(ths, rs, parab, shading='nearest')

Answer 1

参数 shading='nearest' 在极坐标投影中看起来不受支持。 我使用其他网格 xs_ ys_ 手动将颜色 C[i,j] 置于 (ths[i,j], rs[i,j]) 上

fig, ax = plt.subplots(subplot_kw={'projection': 'polar'})

n = 50

xs, ys = np.meshgrid(np.linspace(-1,1,n),np.linspace(-1,1,n))
dx, dy = 0.5*np.ptp(xs)/(n-1), 0.5*np.ptp(ys)/(n-1)
xs_,ys_ = np.meshgrid(np.linspace(-1-dx,1+dx,n+1),np.linspace(-1-dx,1+dx,n+1))

parab = (xs_+dx)**2+(ys_+dx)**2

rs = np.sqrt(xs**2+ys**2)
ths = np.arctan2(ys, xs)

rs_ = np.sqrt(xs_**2+ys_**2)
ths_ = np.arctan2(ys_, xs_)

ax.pcolormesh(ths_, rs_, parab)

plt.show()

输出 ：