我試圖在scipy
中使用splev
的幾個點找到樣條函數的衍生物。例如:樣條函數的衍生物:`scipy splev`
import numpy as np
from scipy.interpolate import splprep, splev
import matplotlib.pyplot as plt
# function to normalize each row
def normalized(a, axis=-1, order=2):
l2 = np.atleast_1d(np.linalg.norm(a, order, axis))
l2[l2==0] = 1
return a/np.expand_dims(l2, axis)
# points and spline
pts = np.array([[0,0],[1,1],[2,np.sqrt(2)],[4,2],[9,3]])
tck, u = splprep(pts.T, u=None, k=3, s=0.0, per=0)
# compute new points and derivatives
u_new = np.linspace(u.min(), u.max(), 5*u.shape[0])
x_new, y_new = splev(u_new, tck, der=0)
xp_num, yp_num = splev(pts, tck, der=1) # numerical derivatices
xp_the, yp_the= pts[1:,0], 0.5/np.sqrt(pts[1:,0]) # analytical derivatives
R = normalized(yp_num/xp_num)
X,Y = pts[1:,0], pts[1:,1]
U,V = X + R[1:,0], Y+R[1:,1]
我想在指定點繪製切線:
plt.plot(x_new,y_new,'-b')
plt.plot(pts[:,0],pts[:,1],'--ro')
plt.quiver(X,Y,U,V, angles='xy', scale_units='xy')
我覺得這些切線都是錯誤的。我的理解是,xp_num
和yp_num
是相對於x
和y
的樣條的數值導數。所以要找到dy/dx
,我應該簡單地將它們分開。它是否正確?
最後,我想找到一個曲線的切線像this
wh在「正常化」嗎? –
@PaulPanzer:它規範化每一行。我加了'normalize'的定義。 – Mahdi
奇怪的是,你的代碼在我看來好像它不應該在劇情中創建任何向量。 'yp_num'和'xp_num'是1d,不是嗎?所以你的'R'應該是1x東西,所以當你切割R [1:,0]時,你應該得到一個空數組。我在這裏錯過了什麼嗎? –