python绘制立体玫瑰花

1.需要的程序包

matplotlib 
numpy

2.代码

from matplotlib import cm
import matplotlib.pyplot as plt
import numpy as np
 
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))
surf = ax.plot_surface(r*np.cos(t), r*np.sin(t), u*(x*np.cos(p)-y*np.sin(p)), rstride=1,
    cstride=1,cmap=cm.Reds_r,linewidth = 0, antialiased = True)#根据参数绘图
 
#自己设置文字内夸和显示设置
ax.text(-0.3,0,1.2,'LOVE', color='red')
plt.axis('off')
plt.show()

3.运行结果