python - 蝴蝶曲线与不同层次的翅膀所有褶皱有不同的颜色

Question

我用以下代码绘制了蝴蝶曲线：

def butterfly_curve(timestamps):
    x=np.zeros(timestamps.size)
    y=np.zeros(timestamps.size)
    
    for k in range(0,timestamps.size):
        x[k] = np.sin(timestamps[k])*(np.exp(np.cos(timestamps[k]))-2*np.cos(4*timestamps[k])-(np.sin((timestamps[k]/12)))**5)
        y[k] = np.cos(timestamps[k])*(np.exp(np.cos(timestamps[k]))-2*np.cos(4*timestamps[k])-(np.sin((timestamps[k]/12)))**5)
    
    return x, y

time = np.arange(0, 12 * np.pi, 0.01)
plt.plot(butterfly_curve(time)[0],butterfly_curve(time)[1])
plt.xlabel('x')
plt.ylabel('y')
plt.axhline(0, color='black')
plt.axvline(0, color='black')
plt.title('Butterfly Curve')

plt.show()

谁能告诉我如何实现不同级别的翅膀的所有褶皱都有不同的颜色？

Answer 1

您的代码复杂且效率低下，该函数可以简单地编写为

In [8]: def butterfly_curve(t):
   ...:     c, s = np.cos(t), np.sin(t)
   ...:     r = (np.exp(c)-2*np.cos(4*t)-(np.sin((t/12)))**5)
   ...:     return s*r, c*r

接下来，要绘制具有不同颜色的“折叠”，您可以使用以下变体（我不知道什么是“折叠”），可能会改变①号。迭代次数，②第一个间隔的长度，③循环结束时的加法常数

In [9]: fig, ax = plt.subplots()
   ...: t = np.linspace(0, np.pi, 301)
   ...: for n in range(12):
   ...:     ax.plot(*butterfly_curve(t))
   ...:     t += np.pi

（请注意，您不需要调用该函数两次，*作为一元前缀运算符是解包运算符）。