ProE常用曲线方程：Python Matplotlib 版本代码（蝴蝶曲线）

         花纹的生成可以使用贴图的方式，同样也可以使用方程，本文列出了几种常用曲线的方程式，以取代贴图方式完成特定花纹的生成。

         注意极坐标的使用.................

         前面部分基础资料，参考：Python：Matplotlib 画曲线和柱状图（Code）

         Pyplot教程：https://matplotlib.org/gallery/index.html#pyplots-examples 

         


        顾名思义，蝴蝶曲线（Butterfly curve ）就是曲线形状如同蝴蝶。蝴蝶曲线如图所示，以方程描述，是一条六次平面曲线。如果大家觉得这个太过简单，别着急，还有第二种。如图所示，以方程描述，这是一个极坐标方程。通过改变这个方程中的变量θ，可以得到不同形状与方向的蝴蝶曲线。如果再施以复杂的组合和变换，我们看到的就完全称得上是一幅艺术品了。

       

Python代码：

import numpy as np
import matplotlib.pyplot as plt
import os,sys,caffe

import matplotlib as mpl
from mpl_toolkits.mplot3d import Axes3D  #draw lorenz attractor
# %matplotlib inline
from math import sin, cos, pi
import math

def mainex():
#drawSpringCrurve();#画柱坐标系螺旋曲线
#HelicalCurve();#采用柱坐标系#尖螺旋曲线
#Votex3D();
#phoenixCurve();
#ButterflyCurve();
#ButterflyNormalCurve();
#dicareCurve2d();
#WindmillCurve3d();
#HelixBallCurve();#球面螺旋线
#AppleCurve();
#HelixInCircleCurve();#使用scatter，排序有问题
seperialHelix();

def drawSpringCrurve():
#碟形弹簧
#圓柱坐标
#方程：
#import matplotlib as mpl
#from mpl_toolkits.mplot3d import Axes3D
#import numpy as np
#import matplotlib.pyplot as plt
mpl.rcParams['legend.fontsize'] = 10;

fig = plt.figure();
ax = fig.gca(projection='3d');

# Prepare arrays x, y, z
#theta = np.linspace(-4 * np.pi, 4 * np.pi, 100)
#z = np.linspace(-2, 2, 100)
#r = z**2 + 1

t = np.arange(0,100,1);
r = t*0 +20;
theta = t*3600 ;

z = np.arange(0,100,1);
for i in range(100):
z[i] =(sin(3.5*theta[i]-90))+24*t[i];

x = r * np.sin(theta);
y = r * np.cos(theta);

ax.plot(x, y, z, label='SpringCrurve');
ax.legend();

plt.show();

def HelicalCurve():
#螺旋曲线#采用柱坐标系
t = np.arange(0,100,1);
r =t ;
theta=10+t*(20*360);
z =t*3;

x = r * np.sin(theta);
y = r * np.cos(theta);

mpl.rcParams['legend.fontsize'] = 10;
fig = plt.figure();
ax = fig.gca(projection='3d');

ax.plot(x, y, z, label='HelicalCurve');
ax.legend();

plt.show();

def ButterflyCurve():
#蝶形曲线，使用球坐标系#或许公式是错误的，应该有更加复杂的公式
t = np.arange(0,4,0.01);

r = 8 * t;
theta = 3.6 * t * 2*1 ;
phi   = -3.6 * t * 4*1;

x = t*1;
y = t*1;
#z = t*1;
z =0
for i in range(len(t)):
x[i] = r[i] * np.sin(theta[i])*np.cos(phi[i]);
y[i] = r[i] * np.sin(theta[i])*np.sin(phi[i]);
#z[i] = r[i] * np.cos(theta[i]);
mpl.rcParams['legend.fontsize'] = 10;
fig = plt.figure();
ax = fig.gca(projection='3d');

ax.plot(x, y, z, label='ButterflyCurve');
#ax.scatter(x, y, z, label='ButterflyCurve');
ax.legend();

plt.show();

def ButterflyNormalCurve():
#蝶形曲线，使用球坐标系#或许公式是错误的，应该有更加复杂的公式
#螺旋曲线#采用柱坐标系
#t = np.arange(0,100,1);

theta=np.arange(0,6,0.1);#(0,72,0.1);
r =theta*0;
z =theta*0;

x =theta*0;
y =theta*0;
for i in range(len(theta)):
r[i] = np.power(math.e,sin(theta[i]))- 2*cos(4*theta[i])
+ np.power( sin(1/24 * (2*theta[i] -pi ) ) , 5 );
#x[i] = r[i] * np.sin(theta[i]);
#y[i] = r[i] * np.cos(theta[i]);
x = r * np.sin(theta);
y = r * np.cos(theta);
mpl.rcParams['legend.fontsize'] = 10;
fig = plt.figure();
ax = fig.gca(projection='3d');

ax.plot(x, y, z, label='ButterflyNormalCurve');
ax.legend();

plt.show();

def phoenixCurve():
#蝶形曲线，使用球坐标系
t = np.arange(0,100,1);

r = 8 * t;
theta = 360 * t * 4 ;
phi   = -360 * t * 8;

x = t*1;
y = t*1;
z = t*1;
for i in range(len(t)):
x[i] = r[i] * np.sin(theta[i])*np.cos(phi[i]);
y[i] = r[i] * np.sin(theta[i])*np.sin(phi[i]);
z[i] = r[i] * np.cos(theta[i]);
mpl.rcParams['legend.fontsize'] = 10;
fig = plt.figure();
ax = fig.gca(projection='3d');

ax.plot(x, y, z, label='phoenixCurve');
ax.legend();

plt.show();

def dicareCurve2d():

r = np.arange(0, 2, 0.01)
theta = 2 * np.pi * r

ax = plt.subplot(111, projection='polar')
ax.plot(theta, r)
ax.set_rmax(2)
ax.set_rticks([0.5, 1, 1.5, 2])  # Less radial ticks
ax.set_rlabel_position(-22.5)  # Move radial labels away from plotted line
ax.grid(True)

ax.set_title("dicareCurve2d", va='bottom')
plt.show();

def WindmillCurve3d():
#风车曲线
t = np.arange(0,2,0.01);
r =t*0+1 ;

#r=1
ang =36*t;#ang =360*t;
s =2*pi*r*t;

x = t*1;
y = t*1;
for i in range(len(t)):
x[i] = s[i]*cos(ang[i]) +s[i]*sin(ang[i]) ;
y[i] = s[i]*sin(ang[i]) -s[i]*cos(ang[i]) ;

z =t*0;

mpl.rcParams['legend.fontsize'] = 10;
fig = plt.figure();
ax = fig.gca(projection='3d');

ax.plot(x, y, z, label='WindmillCurve3d');
ax.legend();

plt.show();

def HelixBallCurve():
#螺旋曲线，使用球坐标系
t = np.arange(0,2,0.005);
r =t*0+4 ;
theta =t*1.8
phi =t*3.6*20

x = t*1;
y = t*1;
z = t*1;
for i in range(len(t)):
x[i] = r[i] * np.sin(theta[i])*np.cos(phi[i]);
y[i] = r[i] * np.sin(theta[i])*np.sin(phi[i]);
z[i] = r[i] * np.cos(theta[i]);
mpl.rcParams['legend.fontsize'] = 10;
fig = plt.figure();
ax = fig.gca(projection='3d');

ax.plot(x, y, z, label='HelixBallCurve');
ax.legend();

plt.show();

def seperialHelix():
#螺旋曲线，使用球坐标系
t = np.arange(0,2,0.1);
n = np.arange(0,2,0.1);
r =t*0+4 ;
theta =n*1.8 ;
phi =n*3.6*20;

x = t*0;
y = t*0;
z = t*0;
for i in range(len(t)):
x[i] = r[i] * np.sin(theta[i])*np.cos(phi[i]);
y[i] = r[i] * np.sin(theta[i])*np.sin(phi[i]);
z[i] = r[i] * np.cos(theta[i]);

mpl.rcParams['legend.fontsize'] = 10;
fig = plt.figure();
ax = fig.gca(projection='3d');

ax.plot(x, y, z, label='ButterflyCurve');
ax.legend();

plt.show();

def AppleCurve():
#螺旋曲线
t = np.arange(0,2,0.01);

l=2.5
b=2.5
x =  t*1;
y =  t*1;
z =0;#z=t*0;
n = 36
for i in range(len(t)):
x[i]=3*b*cos(t[i]*n)+l*cos(3*t[i]*n)
y[i]=3*b*sin(t[i]*n)+l*sin(3*t[i]*n)

#x = r * np.sin(theta);
#y = r * np.cos(theta);

mpl.rcParams['legend.fontsize'] = 10;
fig = plt.figure();
ax = fig.gca(projection='3d');

ax.plot(x, y, z, label='AppleCurve');
ax.legend();

plt.show();

def HelixInCircleCurve():
#园内螺旋曲线#采用柱坐标系
t = np.arange(-1,1,0.01);

theta=t*36 ;#360 deta 0.005鸟巢网 #36 deta 0.005 圆内曲线
x =  t*1;
y =  t*1;
z =  t*1;
r =  t*1;
n = 1.2
for i in range(len(t)):
r[i]=10+10*sin(n*theta[i]);
z[i]=2*sin(n*theta[i]);
x[i] = r[i] * np.sin(theta[i]);
y[i] = r[i] * np.cos(theta[i]);

mpl.rcParams['legend.fontsize'] = 3;
fig = plt.figure();
ax = fig.gca(projection='3d');

ax.plot(x, y, z, label='HelixInCircleCurve');
#ax.scatter(x, y, z, label='HelixInCircleCurve');
ax.legend();

plt.show();

def Votex3D():

def midpoints(x):
sl = ()
for i in range(x.ndim):
x = (x[sl + np.index_exp[:-1]] + x[sl + np.index_exp[1:]]) / 2.0
sl += np.index_exp[:]
return x

# prepare some coordinates, and attach rgb values to each
r, g, b = np.indices((17, 17, 17)) / 16.0
rc = midpoints(r)
gc = midpoints(g)
bc = midpoints(b)

# define a sphere about [0.5, 0.5, 0.5]
sphere = (rc - 0.5)**2 + (gc - 0.5)**2 + (bc - 0.5)**2 < 0.5**2

# combine the color components
colors = np.zeros(sphere.shape + (3,))
colors[..., 0] = rc
colors[..., 1] = gc
colors[..., 2] = bc

# and plot everything
fig = plt.figure();
ax = fig.gca(projection='3d');
ax.voxels(r, g, b, sphere,
facecolors=colors,
edgecolors=np.clip(2*colors - 0.5, 0, 1),  # brighter
linewidth=0.5);
ax.set(xlabel='r', ylabel='g', zlabel='b');
plt.show();

def drawFiveFlower():
theta=np.arange(0,2*np.pi,0.02)
#plt.subplot(121,polar=True)
#plt.plot(theta,2*np.ones_like(theta),lw=2)
#plt.plot(theta,theta/6,'--',lw=2)
#plt.subplot(122,polar=True)
plt.subplot(111,polar=True)
plt.plot(theta,np.cos(5*theta),'--',lw=2)
plt.plot(theta,2*np.cos(4*theta),lw=2)
plt.rgrids(np.arange(0.5,2,0.5),angle=45)
plt.thetagrids([0,45,90]);

plt.show();

if __name__ == '__main__':
import argparse
mainex();

画图结果：