首页 > 解决方案 > 更准确的曲线拟合

问题描述

我通过在 scipy 中使用 curve_fit 函数进行了曲线拟合。但是,曲线拟合对我来说并不好。有什么办法可以改善曲线拟合?下面的python代码是我写的。

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

minus_eta_total =[-0.004, -0.0116, -0.02604, -0.04, -0.08, -0.12, -0.16, -0.2, -0.24, -0.288, -0.3456, -0.4]

I_d_infil_0_sub_1 = [0.004204675, 0.012262849, 0.028331318, 0.045626793, 0.113222124, 0.224759087, 0.403571293, 0.678854397, 1.090002487, 1.837299526, 3.260471157, 5.311900419]
ASR_el = 0.0075

eta_infil_0_sub_1 = np.array(minus_eta_total) + (np.array(I_d_infil_0_sub_1)*np.array(ASR_el))

cons_eq = 8.3144 * 1073 / (0.5 * 4 * 96485)

def func(x, a, b):
    return -cons_eq*np.log(x/a)-b*x

popt_infil_0_sub_1, pcov_infil_0_sub_1 = curve_fit(func, I_d_infil_0_sub_1, eta_infil_0_sub_1)

plt.clf()

plt.plot(I_d_infil_0_sub_1, eta_infil_0_sub_1, linestyle = '--',  marker='o', color='k', label = 'original')
plt.plot(I_d_infil_0_sub_1, func(np.asarray(I_d_infil_0_sub_1),*popt_infil_0_sub_1), 'k', label='fit: $\mathit{j_0}$=%5.4f, R$_{ohm}$=%5.4f' % tuple(popt_infil_0_sub_1))

plt.ylim(-0.42, 0.02)
plt.xticks(np.arange(0, 12, 2))
plt.yticks(np.arange(0, -0.42, -0.05))
plt.xlabel('$\mathit{j}$ ($A/cm^2$)', fontsize=14)
plt.ylabel('-\u03b7$_c$ (V)', fontsize=14)  # \u03bcm = micro(\u03bc) + meter(m)
plt.legend(loc='upper right')
plt.show(block = False)

标签: pythonscipycurve-fitting

解决方案


拟合不如预期的好,可能是因为选择的方程不太方便。

例如,如果我们选择等式:

y(x)=(exp(-a*x)-1)/a

和拟合标准:最小均方相对误差,

a=10.17

结果是 MSRE=0.052

考虑到只有一个参数的如此简单的方程,这是一个非常好的结果。

当然,更精确的拟合当然可以通过选择更复杂、参数更可调的方程来实现。

在此处输入图像描述


推荐阅读