如何在python曲面图上更改颜色并添加网格线?
回答问题 我是新来的,请温柔一点。我正在做一个需要我绘制八极图的项目。我使用了python表面图,得到了这样的结果: 但是,我的教授想要这样的东西: 请忽略图中的226Ra。我想将绘图的颜色更改为红色,并添加这样的网格线。有什么办法可以做到吗?我将不胜感激。这是我的代码: import numpy as np; from matplotlib import cm, colors; import c
·
回答问题
我是新来的,请温柔一点。我正在做一个需要我绘制八极图的项目。我使用了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);
Answers
只需将您的plot_surface
行替换为以下内容并注释掉fig.colorbar(m, shrink=0.8);
。您可以根据自己的意愿使用这个页面中的任何其他颜色。
ax.plot_surface(X, Y, Z, rstride=10, cstride=10, color='orangered', edgecolors='k', lw=0.6)
输出
更多推荐
已为社区贡献126473条内容
所有评论(0)