在Python中用方程两侧有因变量的给定方程拟合数据

import numpy as np
from scipy.optimize import curve_fit

def quadratic_func(x, a, b, c):
    return a * x**2 + b * x + c

x_data = np.array([1, 2, 3, 4, 5])
y_data = np.array([2.1, 4.9, 8.2, 15.7, 25.5])

params, params_covariance = curve_fit(quadratic_func, x_data, y_data)

a = params[0]
b = params[1]
c = params[2]

x_new = np.array([6, 7, 8])
y_new = quadratic_func(x_new, a, b, c)