封闭曲线拟合python 代码

以下是一个基于Python的封闭曲线拟合代码示例：

import numpy as np
from scipy.optimize import curve_fit

# 定义一个椭圆形函数
def ellipse_func(x, a, b, h, k, theta):
    theta = np.radians(theta)
    x1 = (x[0]-h)*np.cos(theta) + (x[1]-k)*np.sin(theta)
    x2 = (x[0]-h)*np.sin(theta) - (x[1]-k)*np.cos(theta)
    return a*np.cos(np.arctan2(x2/b, x1/a))

# 生成模拟数据
x_data = np.linspace(0, 2*np.pi, 100)
y_data = 2*np.sin(x_data) + 1*np.cos(2*x_data) + np.random.normal(0, 0.1, size=len(x_data))

# 初始参数猜测
init_guess = (2, 1, 0, 0, 45)

# 使用最小二乘法进行参数拟合
popt, pcov = curve_fit(ellipse_func, (x_data, y_data), y_data, p0=init_guess)

# 输出拟合结果
print('a =', popt[0])
print('b =', popt[1])
print('h =', popt[2])
print('k =', popt[3])
print('theta =', popt[4])

这个代码示例中，我们使用 `scipy.optimize` 库中的 `curve_fit` 函数来拟合一个椭圆形函数，该函数的参数包括：椭圆的长轴和短轴长度、椭圆的中心坐标、以及椭圆的旋转角度。我们使用最小二乘法来拟合这个函数，并输出拟合结果。

你可以根据需要修改函数 `ellipse_func`，以适应不同的曲线拟合需求。