[英]Animation of tangent line of a 3D curve
我正在編寫一個 Python 程序來為沿 3D 曲線的切線設置動畫。 但是,我的切線沒有移動。 我認為問題是
line.set_data(np.array(Tangent[:,0]).T,np.array(Tangent[:,1]).T)
在animate(i)
但我無法弄清楚。 任何幫助將不勝感激。 以下是代碼。
from mpl_toolkits import mplot3d
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
import matplotlib
matplotlib.use( 'tkagg' )
plt.style.use('seaborn-pastel')
fig = plt.figure()
ax = plt.axes(projection='3d')
ax = plt.axes(projection='3d')
# Data for a three-dimensional line
zline = np.linspace(0, 15, 1000)
xline = np.sin(zline)
yline = np.cos(zline)
ax.plot3D(xline, yline, zline, 'red')
def curve(t):
return [np.sin(t),np.cos(t),t]
def vector_T(t):
T = [np.cos(t),-np.sin(t),1]
return T/np.linalg.norm(T)
len = 2
def tangent_line(t):
P = np.add(curve(t),len*vector_T(t))
Q = np.subtract(curve(t),len*vector_T(t))
return np.array([P, Q]).T
t0 = 0
Tangent=tangent_line(t0)
line, = ax.plot3D(Tangent[0], Tangent[1], Tangent[2], 'green')
def init():
line.set_data([], [])
return line,
def animate(i):
t0 = 15* (i/200)
Tangent=tangent_line(t0)
#print(Tangent)
line.set_data(np.array(Tangent[:,0]).T,np.array(Tangent[:,1]).T)
return line,
anim = FuncAnimation(fig, animate, init_func=init,
frames=200, interval=20, blit=True)
plt.show()
您在animate
調用了錯誤的函數:將line.set_data(...)
替換為line.set_data_3d(Tangent[0], Tangent[1], Tangent[2])
它將起作用。
代碼中仍然存在一些小問題(例如,不要使用len
作為變量名)。 我建議使用以下內容:
#!/usr/bin/env python3
from mpl_toolkits import mplot3d
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
import matplotlib
matplotlib.use('tkagg')
plt.style.use('seaborn-pastel')
fig = plt.figure()
ax = plt.axes(projection='3d')
# Data for a three-dimensional line
zline = np.linspace(0, 15, 1000)
xline = np.sin(zline)
yline = np.cos(zline)
ax.plot3D(xline, yline, zline, 'red')
def curve(t):
return [ np.sin(t), np.cos(t), t ]
def tangent(t):
t = [ np.cos(t), -np.sin(t), 1.0 ]
return t/np.linalg.norm(t)
def tangent_line(t):
length = 2.0
offset = length * tangent(t)
pos = curve(t)
return np.array([ pos-offset, pos+offset ]).T
line = ax.plot3D(*tangent_line(0), 'green')[0]
def animate(i):
line.set_data_3d(*tangent_line(15* (i/200)))
return [ line ]
anim = FuncAnimation(fig, animate, frames=200, interval=20, blit=True)
plt.show()
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.