python - Python中网格数据的PolarColor map 图

我想绘制极坐标形式的网格数据的颜色图。在数据中，r的范围是1到2，theta 0到360。我想要这样的映射：


    我这样画

#x1 of shape(12,)
#y1 0f shape(36,1)
#z of shape(36,12)
fig=plt.figure()
ax=fig.add_subplot(111) #Output Figure 1
#ax=fig.add_subplot(111,polar='True') #Output Figure 2
ax=fig.add_subplot(111)
ax.pcolor(y1,x1,z)
plt.show()

输出：




知道如何绘制上图吗？我还尝试将r，theta转换为x，y，然后我得到了我不想要的r

最佳答案

正如here所指出的，pcolormesh()对此很有用。

import matplotlib.pyplot as plt
import numpy as np

theta = np.linspace(0, 2*np.pi, 36)
r = np.linspace(1, 2, 12)

R, THETA = np.meshgrid(r, theta)
Z = np.sin(THETA) * R

plt.subplot(111, projection='polar')
plt.pcolormesh(THETA, R, Z)
plt.gca().set_rmin(0.0)

plt.show()

output figure