python--polar极坐标热力图

import numpy as np
import matplotlib.pyplot as plt           #导入数据库

theta=np.linspace(0, 360, 30)/180*np.pi
r=np.linspace(0,500,100)

r,theta=np.meshgrid(r,theta)            #创建网格

Z = np.random.rand(30,100)       #随机生成填入网格的数据，根据网格随机生成一个30,100的矩阵

fig = plt.figure(figsize=(10,10))        #创建画布

ax = plt.subplot(projection='polar')
ax.set_rgrids([100,200,300,400])         #创建极坐标系，插入点

im = ax.contourf(theta, r , Z,cmap = plt.cm.RdGy)        #绘图

fig.colorbar(im,ax=ax)      #显示colorbar

plt.show()    #显示图片