Matplotlib多曲线拟合报错：curve_fit参数p0重复值问题求助

错误根源 #

TypeError: curve_fit() got multiple values for argument 'p0' 是因为调用scipy.optimize.curve_fit()时，p0参数被同时通过位置参数和关键字参数传递，或者参数顺序不符合函数定义导致的。curve_fit的标准调用签名是：

curve_fit(func, xdata, ydata, p0=None, ...)

如果你的代码里既在位置上传递了初始参数，又用p0=指定，就会触发这个错误。

修正方案与示例代码 #

假设你的自定义erf拟合函数为：

from scipy.special import erf
import numpy as np
from scipy.optimize import curve_fit

# 基于erf的拟合模型
def erf_model(x, a, b, mu, sigma):
    return a * erf((x - mu)/(sigma * np.sqrt(2))) + b

针对三条实验曲线的拟合，修正后的完整代码如下：

import matplotlib.pyplot as plt

# 替换为你的真实实验数据
x_data = np.linspace(0, 10, 100)
y1_data = 2 * erf((x_data - 5)/(1.5 * np.sqrt(2))) + 1 + np.random.normal(0, 0.1, 100)
y2_data = 1.5 * erf((x_data - 4)/(2 * np.sqrt(2))) + 0.5 + np.random.normal(0, 0.1, 100)
y3_data = 3 * erf((x_data - 6)/(1 * np.sqrt(2))) + 2 + np.random.normal(0, 0.1, 100)

# 每组曲线的初始参数猜测（根据数据趋势调整）
initial_guesses = [
    [2, 1, 5, 1.5],
    [1.5, 0.5, 4, 2],
    [3, 2, 6, 1]
]
y_datasets = [y1_data, y2_data, y3_data]
curve_labels = ["曲线1", "曲线2", "曲线3"]

plt.figure(figsize=(10,6))

for y_data, guess, label in zip(y_datasets, initial_guesses, curve_labels):
    # 正确调用：仅用关键字参数传递p0，或作为位置参数（二选一）
    popt, pcov = curve_fit(erf_model, x_data, y_data, p0=guess)
    a_fit, b_fit, mu_fit, sigma_fit = popt
    
    # 计算R²拟合优度
    y_pred = erf_model(x_data, *popt)
    ss_res = np.sum((y_data - y_pred)**2)
    ss_tot = np.sum((y_data - np.mean(y_data))**2)
    r_squared = 1 - (ss_res / ss_tot)
    
    # 输出拟合结果
    print(f"{label}拟合参数：")
    print(f"a={a_fit:.4f}, b={b_fit:.4f}, mu={mu_fit:.4f}, sigma={sigma_fit:.4f}")
    print(f"R²={r_squared:.4f}\n")
    
    # 绘图
    plt.plot(x_data, y_data, 'o', markersize=3, label=f"{label}实验数据")
    plt.plot(x_data, y_pred, '-', label=f"{label}拟合曲线 (R²={r_squared:.4f})")

plt.xlabel("X轴")
plt.ylabel("Y轴")
plt.legend()
plt.show()

核心注意事项 #

• 调用curve_fit时，p0只能传递一次：要么紧跟在ydata后面作为位置参数，要么用p0=xxx关键字参数，禁止同时使用两种方式。
• 初始参数猜测p0要尽量接近实际数据的趋势，否则可能导致拟合不收敛或结果失真。
• R²值越接近1，说明拟合效果越好。

内容的提问来源于stack exchange，提问作者KeplerNick123