如何在python中创建N个指数函数来拟合数据

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

# 定义一个N个指数函数的通用形式
def multi_exponential(x, *params):
    y = np.zeros_like(x)
    for i in range(0, len(params), 3):  # 每三个参数为一个指数函数
        a, b, c = params[i:i+3]
        y += a * np.exp(-b * x) + c
    return y

# 生成一些模拟数据
x_data = np.linspace(0, 4, 50)
y_data = multi_exponential(x_data, 2.5, 1.3, 0.5, -1.5, 0.5, 1.5) + 0.2 * np.random.normal(size=len(x_data))

# 初始参数猜测
initial_guess = [1, 1, 0, 1, 1, 0]  # 根据实际情况调整

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

# 输出拟合参数
print('Fitted parameters:', params)

# 计算拟合曲线
y_fit = multi_exponential(x_data, *params)

# 绘制结果
plt.scatter(x_data, y_data, label='Data')
plt.plot(x_data, y_fit, label='Fit', color='red')
plt.legend()
plt.show()