Matplotlib polar contourf plot：跨 theta 原点连续

Question

我的数据格式为 E(freq, theta)，其中 E 是二维数组，freq 和 theta 是一维数组。

以下代码部分生成附图。但是，我想让 contourf 图在 0 度原点上连续（即沿 0 方位角没有白色楔形 space）。

我已经研究了 matplotlib 文档，并非常广泛地发布了问题，但似乎无法找到解决此问题的方法。有什么想法吗？

代码：

[r, th] = np.meshgrid(freq,theta)

fig = plt.figure()
ax = fig.add_subplot(111, polar=True)
ax.set_theta_zero_location('N')
ax.set_theta_direction(-1)

cntf = ax.contourf(th,r,np.log10(E),cmap='jet',extend='both',
    levels=np.linspace(np.mean(np.log10(E)), np.amax(np.log10(E)), 15))

ax.set_rlim(0, .3)
label_position=ax.get_rlabel_position()
ax.text(np.radians(label_position+25),ax.get_rmax()/1.5,'f (Hz)',
        rotation=label_position,ha='center',va='center')

制作剧情：

Answer 1

类似于此：

dtheta = np.diff(theta).mean()
wrp_theta = np.concatenate((theta, theta[-1:] + dtheta))
wrp_E = np.concatenate((E, E[0:1, :]), axis=0)

Matplotlib polar contourf plot：跨 theta 原点连续

Matplotlib polar contourf plot: continuous across theta origin

python

matplotlib

plot

polar-coordinates

contourf