python图像加网格_如何在python曲面图上改变颜色和添加网格线？

我是新来的，请温柔一点。

我正在做一个项目，需要我画八极体的图形。我使用python surface plot得到如下结果：

然而，我的教授想要这样的东西：

请忽略图中的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);