python对一组数据进行椭圆拟合，并求出长短轴的实现代码

以下是Python对一组数据进行椭圆拟合，并求出长短轴的示例代码：

import numpy as np
from scipy.optimize import curve_fit

def ellipse_func(x, xc, yc, rx, ry, phi):
    x0 = x[:, 0] - xc
    y0 = x[:, 1] - yc
    X = x0 * np.cos(phi) + y0 * np.sin(phi)
    Y = -x0 * np.sin(phi) + y0 * np.cos(phi)
    return ((X / rx) ** 2 + (Y / ry) ** 2 <= 1)

def fit_ellipse(x_data, y_data):
    x0 = (x_data.min() + x_data.max()) / 2
    y0 = (y_data.min() + y_data.max()) / 2
    xc, yc = x0, y0
    guess = [xc, yc, 1, 1, 0]
    popt, pcov = curve_fit(ellipse_func, np.array([x_data, y_data]).T, np.ones(len(x_data)), p0=guess)
    a, b = popt[2], popt[3]
    return a, b

# 示例
x_data = np.array([1, 2, 3, 4, 5])
y_data = np.array([1, 3, 5, 7, 9])
a, b = fit_ellipse(x_data, y_data)
print("长轴: ", 2 * a)
print("短轴: ", 2 * b)

此代码将一组二维坐标数据(x_data, y_data)拟合为一个椭圆，并返回该椭圆的长轴和短轴，并打印出结果。注意：此代码仅作示例，具体使用需根据实际情况进行调整。