两条曲线的关节拟合（python代码）

简介： 对于一组数据，我们可以使用一个简单的线性回归进行拟合；而对于多组数据，我们可以进行联合线性回归（实验参数对于每组数据拟合都是相同的）。
部分代码：

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

y1 = np.array([ 0.01, 0.02, 0.03, 0.04 , 0.05, 0.06,0.07])
y2 = np.array([0.002, 0.067, 0.017, 0.019, 0.023,0.046])
comboY = np.append(y1, y2)
print("comboy:",comboY)
x1 = np.array([5,6, 7, 8, 9,10,11])
x2 = np.array([6,7, 8, 9,10,11])
comboX = np.append(x1, x2)
print("comboX:",comboX)
if len(y1) != len(x1):
 raise(Exception('Unequal x1 and y1 data length'))
if len(y2) != len(x2):
 raise(Exception('Unequal x2 and y2 data length'))
def function1(data, a, b, c):
 return a * data + c
def function2(data, a, b, c):
 return b * data + c
def combinedFunction(comboData, a, b, c):
    extract1 = comboData[:len(x1)]
    print("extract1:", extract1)
    result1 = function1(extract1, a, b, c)
    print("result1:", result1)

    extract2 = comboData[len(x1):]
    result2 = function2(extract2, a, b, c)
    return np.append(result1, result2)
initialParameters = np.array([1.0, 1.0, 1.0])
fittedParameters, pcov = curve_fit(combinedFunction, comboX, comboY, initialParameters)
......

实验结果

完整代码链接：两条曲线的关节拟合（python代码）