Python求图的给定坐标方程

Question

我试图建立一个程序，找到（粗略地）一个图的方程，给定一组点在图上的坐标。假设给定的坐标代表整个图（即：行为在给定域外不会改变）。在

我使用下面的简单函数来执行此操作：

#x and y are arrays of the x coordinates and corrosponding y coordinates
def get_equation(x,y):
    degree = 2
    coefs, res, _, _, _ = np.polyfit(x,y,degree, full = True)
    ffit = np.poly1d(coefs)
    print (ffit)
    return ffit

对于基本方程图（如x^2）上的坐标，这种方法相当有效，但对于更复杂的图形（如下图）则根本不起作用。在

如何找到更复杂的图的方程，比如上面给出的图的坐标？在

另外，有没有可能计算出图形的度数，或者总是需要手动输入？在

Answer 1

如果行为在给定的域外没有改变，请查看样条曲线并将其与域相匹配。这可以用scipy.插值. 在

这里有一个例子

 from matplotlib.pyplot import subplots
from numpy import linspace, random, sin, cos
from scipy import interpolate

x = linspace(0, 10)

y = sin(x * .5) + cos (x * 2)  + random.randn(x.size) * 1e-1
# fit spline
spl = interpolate.InterpolatedUnivariateSpline(x, y)
fitx = linspace(0, x.max(), 100)

fig, ax = subplots()
ax.scatter(x, y)

ax.plot(fitx, spl(fitx))
fig.show()