mirror of
https://github.com/sun0225SUN/Awesome-Love-Code.git
synced 2026-08-31 01:37:17 +08:00
Crany-web
25 lines
782 B
Python
25 lines
782 B
Python
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
from matplotlib import cm
|
|
|
|
fig = plt.figure()
|
|
ax = fig.add_subplot(projection = '3d')
|
|
[x, t] = np.meshgrid(
|
|
np.array(range(25)) / 24.0,
|
|
np.arange(0, 575.5, 0.5) / 575 * 17 * np.pi - 2 * np.pi)
|
|
p = (np.pi / 2) * np.exp(-t / (8 * np.pi))
|
|
u = 1 - (1 - np.mod(3.6 * t, 2 * np.pi) / np.pi)**4 / 2
|
|
y = 2 * (x**2 - x)**2 * np.sin(p)
|
|
r = u * (x * np.sin(p) + y * np.cos(p))
|
|
h = u * (x * np.cos(p) - y * np.sin(p))
|
|
c = cm.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()
|