极坐标热图成像

import matplotlib.pyplot as plt

from mpl_toolkits.mplot3d import Axes3D

import numpy as np

from matplotlib import colors,ticker,cm

fig = plt.figure()

ax = Axes3D(fig)

rad = np.linspace(0, 5, 100)

azm = np.linspace(0, 2 * np.pi, 100)

r, th = np.meshgrid(rad, azm)

z = (r ** 2.0) / 4.0

plt.subplot(projection="polar")

plt.pcolormesh(th, r, z,cmap=plt.cm.jet)

#plt.pcolormesh(th, z, r)

plt.plot(azm, r, color='k', ls='none')

plt.grid()

plt.show()