python colorbar 0变白色_Python中colorbar中的贴图颜色

我使用下面的代码来生成三维曲面图。我想用颜色来代表半径的值。如何绘制相应的颜色栏？在

我相信问题是我在绘制曲面图时使用了facecolors。所以我需要根据这个脸色做颜色条。在

我知道我必须把这些颜色映射到colorbar，只是我不知道怎么做。在import numpy as np

import matplotlib.pyplot as plt

def sph2cart(theta, phi, radius):

x_cart = radius * np.sin(theta) * np.cos(phi)

y_cart = radius * np.sin(theta) * np.sin(phi)

z_cart = radius * np.cos(theta)

return np.array([x_cart, y_cart, z_cart])

def antenna_plot_3D(mod_F, T, P):

[x, y, z] = sph2cart(T, P, mod_F)

fig = plt.figure()

ax = fig.add_subplot(1, 1, 1, projection='3d')

p_surf = ax.plot_surface(x, y, z, rstride=1, cstride=1,

linewidth=0, antialiased=True,

facecolors=cm.jet(np.sqrt(x*x + y*y + z*z)))

ax.set_zlabel('Z axis', fontsize=15)

ax.set_xlabel('X axis', fontsize=15)

ax.set_ylabel('Y axis', fontsize=15)

plt.savefig('3d_pattern.png')

plt.show()

x = 300

t = np.linspace(0, np.pi, num=x)

p = np.linspace(0,2*np.pi, num=x)

[T,P] = np.meshgrid(t,p)

mod_F = np.sin(T) * np.cos(P)

antenna_plot_3D(mod_F, T, P)

我希望曲面是彩色的，颜色应该代表半径。在