如何将部分余弦曲线拟合到Python中的数据？

import numpy as np
from scipy.optimize import curve_fit
import matplotlib.pyplot as plt

# 定义部分余弦函数模型
def partial_cosine(x, a, b, c, d):
    return a * np.cos(b * x + c) + d

# 生成示例数据
x_data = np.linspace(0, 10, 100)
y_data = partial_cosine(x_data, 2, 1, np.pi/2, 1) + np.random.normal(0, 0.2, 100)

# 使用curve_fit进行拟合
params, params_covariance = curve_fit(partial_cosine, x_data, y_data)

# 打印拟合参数
print("拟合参数：")
print("a =", params[0])
print("b =", params[1])
print("c =", params[2])
print("d =", params[3])

# 绘制原始数据和拟合曲线
plt.scatter(x_data, y_data, label='Data')
plt.plot(x_data, partial_cosine(x_data, params[0], params[1], params[2], params[3]), 'r', label='Fit')
plt.legend()
plt.show()