Python scipy.optimize.curve_fit 拟合函数失败

Question

我正在尝试拟合 LRC 电路传递函数，但 curve_fit 似乎无法拟合。我尝试了不同的初始值以及尝试实现 scipy.optimize.differential_evolution.

以下是我的拟合尝试，在绘图后产生 this。

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

def f(freq, R, L, C):
    w = 2*np.pi*np.array(freq) # Convert frequency to angular frequency
    return R*w*C/np.sqrt((1-C*L*w**2)**2+(C*w*R)**2) # Series Transfer Function

name = ['R', 'L', 'C']
p0 = [1000,0.07,0.0000001] # Actual values of circuit components.

params, covariance = curve_fit(f, np.array(x), np.array(y), p0)
print(params)

我不太确定问题出在我的初始参数上，还是 Levenberg-Marquardt 算法做的事情完全不同。

当我尝试将拟合限制在函数的严格递增部分时，拟合似乎完美无缺，这让我认为我的错误一定与函数的形状有关。

有没有人能告诉我为什么我会变得如此奇怪？

Answer 1

这里是一个使用背后的物理原理进行参数猜测的解决方案：

一个人知道最大值在 'U'。此外，已知1/(L*C)= (2 * pi * fMax )**2。最终我们可以使用 U(f) 近似于 2 * pi * f* R * C 小 f。从这个等式可以很容易地计算出猜测

然而，L和C只能相对于R拟合。因此 R 设置为统一。为了显示这种缩放行为，我制作了第二个图，其中 R 设置为 1000 欧姆。如果 R -> a * R，则 L -> a * L，以及 C -> C / a。相应的图表应该是相同的。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('font',**{'family':'serif','serif':['Times']})
rc('text', usetex=True)
from scipy.optimize import curve_fit

def rlc_series( f, u , l, c, R=1):
    r = R
    w = 2 * np.pi * f
    return u * w * r * c / np.sqrt( ( 1 - w**2 * l * c )**2 + ( w * r * c )**2 )

fuData = np.loadtxt( "ohlVW.txt", delimiter=',' )
fuData = fuData[ fuData[:,0].argsort()]

### r=1
guessU = max( fuData[::,1] )
guessC = 1. / ( 2 * np.pi ) * ( fuData[ 1, 1 ] - fuData[ 0, 1 ] ) / (fuData[ 1, 0 ] - fuData[ 0, 0 ] ) / guessU
guessL = 1. / ( 2 * np.pi * fuData[ fuData[:,1].argmax() ,0 ] )**2 * 1./guessC
guess = [ guessU, guessL, guessC ]

sol, pcov = curve_fit(rlc_series, fuData[::,0], fuData[::,1], p0=guess )
print sol

fList = np.linspace( 0, 4000, 1000 )
uGList = np.fromiter( ( rlc_series( x, *guess ) for x in fList ), np.float )
uFList = np.fromiter( ( rlc_series( x, *sol ) for x in fList ), np.float )
uOtherList = np.fromiter( ( rlc_series( x, sol[0], sol[1]*1000, sol[2] / 1000., R=1000 ) for x in fList ), np.float )

fig = plt.figure()
ax = fig.add_subplot( 1, 1, 1 )
ax.plot( fuData[::,0], fuData[::,1] , marker ='o', ls='', label='Data' )
ax.plot( fList, uGList, label='Guess')
ax.plot( fList, uFList, label='Fit ($R=1\;\Omega$)')
ax.plot( fList, uOtherList, ls='--', label='scaled for $R=1000\;\Omega$')
ax.set_xlabel('$f\;\mathrm{(Hz)}$', fontsize=16)
ax.set_ylabel('$U\;\mathrm{(V)}$', fontsize=16)
ax.legend( loc=0 )
plt.show()

提供：

>> [3.85138538e-01 2.95465209e-04 3.49390461e-05]

Python scipy.optimize.curve_fit 拟合函数失败

Python scipy.optimize.curve_fit Failing to Fit Function

python

physics

curve-fitting

scipy