python画3d樱花 立体 渐变色 七夕表白源码

[python]代码库

import numpy as np

import matplotlib.pyplot as plt

from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()

ax = fig.gca(projection='3d')

[x, t] = np.meshgrid(np.array(range(25)) / 24.0, np.arange(0, 575.5, 0.5) / 575 * 6 * np.pi - 4 * np.pi)

p = (np.pi / 2) * np.exp(-t / (8 * np.pi))

change = np.sin(10 * t) / 20

u = 1 - (1 - np.mod(5.2 * t, 2 * np.pi) / np.pi) ** 4 / 2 + change

y = 2 * (x ** 2 - x) ** 2 * np.sin(p)

r = u * (x * np.sin(p) + y * np.cos(p)) * 1.5

h = u * (x * np.cos(p) - y * np.sin(p))

c = plt.get_cmap('spring_r')

surf = ax.plot_surface(r * np.cos(t), r * np.sin(t), h, rstride=1, cstride=1,

                       cmap=c, linewidth=0, antialiased=True)

plt.show()

[代码运行效果截图]