运行结果展示
代码
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 * 30 * np.pi - 4 * np.pi) p = (np.pi / 2) * np.exp(-t / (8 * np.pi)) change = np.sin(20 * t) / 50 u = 1 - (1 - np.mod(3.3 * 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('gist_heat') surf = ax.plot_surface(r * np.cos(t), r * np.sin(t), h, rstride=1, cstride=1, cmap=c, linewidth=0, antialiased=True) ax.set_xticks([]) ax.set_yticks([]) ax.set_zticks([]) plt.show()
引用来源
链接:作者发表的微信文章
小结
非常优秀的代码作者!!!,关注我,有好玩的代码会继续给大家分享