假设我有一个从点[0,0]到[10,10]的三次样条,其边界值为零导数:
spl = scipy.interpolate.CubicSpline(x = [0,10], y = [0,10], bc_type=((1,0),(1,0)))
如果我将在x处评估的样条曲线的输出作为其y坐标,则会得到从起点到终点的“路径”:
y_coordinate = spl(x_coordinate)
现在很容易评估此样条的规则间隔x,例如使用numpy.linspace。
如果我有兴趣找到x的序列,且序列点之间(x,y)空间中的欧几里得距离相等,该怎么办?
最佳答案
当我希望几件事情相等时,我将它们放在一个数组中,并要求SciPy最小化该数组的方差。它倾向于运作良好。
在下面的代码中,我决定(无论好坏)仅使用内部点作为变量(称为t
),并在计算距离之前插入端点a,b。替代方法是将所有点都用作变量,但强加t[0] == a
和t[-1] == b
的约束。
import numpy as np
from scipy import interpolate, optimize
import matplotlib.pyplot as plt
spl = interpolate.CubicSpline(x = [0, 10], y = [0, 10], bc_type=((1, 0), (1, 0)))
def distances_var(t, a, b):
atb = np.concatenate(([a], t, [b])) # added endpoints
y = spl(atb)
dist_squared = np.diff(atb)**2 + np.diff(y)**2
return np.var(dist_squared)
a, b = 0, 10
n = 7 # how many points to put
res = optimize.minimize(distances_var, np.linspace(a, b, n)[1:-1], args=(a, b))
x = np.concatenate(([a], res.x, [b])) # the points we want
xx = np.linspace(a, b, 500) # for plotting the curve
plt.axes().set_aspect('equal')
plt.plot(xx, spl(xx))
plt.plot(x, spl(x), 'ro')
plt.show()
关于python - Python:沿着三次样条曲线找到x(返回y),这会产生等距离的顺序(x,y)对,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/48424136/