python画网格并填充颜色_如何更改颜色并在python曲面图上添加网格线？

我是新来的,请保持温柔.

我正在做一个项目,要求我画出八极的形象.我用

python表面绘图得到了这样的结果：

但是,我的教授想要这样的事情：

请忽略图中的226Ra.我想将我的绘图颜色更改为红色,并添加这样的网格线.有什么办法可以吗？

我将非常感激.

这是我的代码：

import numpy as np;

from matplotlib import cm, colors;

import cmath;

import matplotlib.pyplot as plt;

from mpl_toolkits.mplot3d import Axes3D;

import scipy.special as sp;

l = 3 # degree

m = 0 # order

PHI, THETA = np.mgrid[0:2*np.pi:300j, 0:np.pi:150j]

R = sp.sph_harm(m, l, PHI, THETA).real

s = 1

X = (s*R*0.28+1) * np.sin(THETA) * np.cos(PHI)

Y = (s*R*0.28+1) * np.sin(THETA) * np.sin(PHI)

Z = (s*R*0.29+1) * np.cos(THETA)

norm = colors.Normalize()

fig, ax = plt.subplots(subplot_kw=dict(projection='3d'), figsize=(14,10))

m = cm.ScalarMappable(cmap=cm.jet)

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, facecolors=cm.jet(norm(R)))

m.set_array(R)

fig.colorbar(m, shrink=0.8);