Поверхность питона постепенно меняется цвет - PullRequest
0 голосов
/ 29 августа 2018

Мне нужно сделать фигуру для проекта. Теперь я получаю 90% этого. Однако я хочу, чтобы у моей фигуры был постепенно изменяющийся цвет, немного красного и черного. Есть ли в Python, где я могу достичь этого?

Вот цифра, которую я получил: The figure I got

Вот фигура, которую я хочу: The figure with some gradually varied color Вот мой код

import numpy as np
from matplotlib import cm, colors
import matplotlib.pyplot as plt
import scipy.special as sp
from mpl_toolkits.mplot3d import Axes3D


phi, theta = np.mgrid[0:2*np.pi:300j, 0:np.pi:150j]
Y30 = sp.sph_harm(0, 3, phi, theta).real
Y20 = sp.sph_harm(0, 2, phi, theta).real
R0 = 1
alpha30 = 0.1
alpha20 = 0.02
prefactor = alpha30*Y30 + alpha20 * Y20 +1
X = R0 * prefactor * np.sin(theta) * np.cos(phi)
Y = R0 * prefactor * np.sin(theta) * np.sin(phi)
Z = R0 * prefactor * np.cos(theta)

norm = colors.Normalize()
fig, ax = plt.subplots(subplot_kw=dict(projection='3d'), figsize=(10.5,10))
m = cm.ScalarMappable(cmap=cm.jet)

# hide the background axes
plt.axis('off')

# change the view
ax.view_init(1, )

ax.plot_surface(X, Y, Z, rstride=10, cstride=10, color = 'orangered', edgecolors='k',shade = True)
m.set_array(R0)

Мне не нужно, чтобы моя фигура была точно такой же. Я просто хочу, чтобы в нем была смесь красного и черного, постепенно меняющаяся. Буду очень признателен за помощь.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...