小编kan*_*kar的帖子

我想绘制一个抛物面f(r)= r**2作为2D极地热图.我期望的输出是

我写的代码是

from pylab import*
from mpl_toolkits.mplot3d import Axes3D
ax = Axes3D(figure())
rad=linspace(0,5,100)
azm=linspace(0,2*pi,100)
r,th=meshgrid(rad,azm)
z=(r**2.0)/4.0
subplot(projection="polar")
pcolormesh(r,th, z)
show()

但是该程序返回以下图像.

有人可以帮忙吗？先感谢您.