python - 通过 scipy 拟合函数传递 numpy 数组

Question

具有拟合函数:

def conv_gauss(x, constant, c, beta, sigma): 
    return constant * np.exp((x-c)/beta) * math.erfc((x-c)/(np.sqrt(2)*sigma) + sigma/(np.sqrt(2)*beta))

以及 x 和 y 数据:

x_data = [5751.0, 5752.0, 5753.0, 5754.0, 5755.0, 5756.0, 5757.0, 5758.0, 5759.0, 5760.0, 5761.0, 5762.0, 5763.0, 5764.0, 5765.0, 5766.0, 5767.0, 5768.0, 5769.0, 5770.0, 5771.0, 5772.0, 5773.0, 5774.0, 5775.0, 5776.0, 5777.0, 5778.0, 5779.0, 5780.0, 5781.0, 5782.0, 5783.0, 5784.0, 5785.0, 5786.0, 5787.0, 5788.0, 5789.0, 5790.0, 5791.0, 5792.0, 5793.0, 5794.0, 5795.0, 5796.0, 5797.0, 5798.0, 5799.0, 5800.0, 5801.0, 5802.0, 5803.0, 5804.0, 5805.0, 5806.0, 5807.0, 5808.0, 5809.0, 5810.0, 5811.0, 5812.0, 5813.0, 5814.0, 5815.0, 5816.0, 5817.0, 5818.0, 5819.0, 5820.0, 5821.0, 5822.0, 5823.0, 5824.0, 5825.0, 5826.0, 5827.0, 5828.0, 5829.0, 5830.0, 5831.0, 5832.0, 5833.0, 5834.0, 5835.0, 5836.0, 5837.0, 5838.0, 5839.0, 5840.0, 5841.0, 5842.0, 5843.0, 5844.0, 5845.0, 5846.0, 5847.0, 5848.0, 5849.0, 5850.0]

y_data = [2250.0, 2259.0, 2382.0, 2546.0, 2527.0, 2837.0, 2972.0, 3154.0, 3345.0, 3664.0, 4209.0, 4415.0, 4857.0, 5372.0, 5985.0, 6743.0, 7735.0, 8634.0, 9555.0, 11085.0, 12155.0, 13598.0, 15205.0, 17236.0, 19170.0, 21515.0, 24252.0, 26325.0, 28877.0, 31945.0, 35266.0, 38525.0, 42205.0, 45128.0, 48527.0, 52116.0, 55999.0, 59361.0, 62160.0, 65897.0, 69085.0, 71548.0, 73684.0, 74978.0, 76676.0, 76531.0, 76203.0, 75874.0, 74414.0, 72638.0, 69968.0, 67469.0, 64681.0, 61530.0, 58005.0, 54193.0, 51032.0, 47377.0, 43618.0, 40237.0, 37268.0, 33845.0, 30856.0, 27464.0, 25015.0, 22483.0, 20294.0, 18180.0, 16023.0, 14171.0, 12534.0, 11074.0, 9764.0, 8708.0, 7672.0, 6668.0, 5988.0, 5208.0, 4585.0, 4163.0, 3845.0, 3326.0, 2904.0, 2784.0, 2529.0, 2271.0, 2219.0, 2103.0, 1943.0, 1766.0, 1782.0, 1650.0, 1578.0, 1576.0, 1481.0, 1483.0, 1357.0, 1365.0, 1338.0, 1269.0]

通过拟合函数:

popt, pcov = optimize.curve_fit(conv_gauss, x_data, y_data, p0=[1,5800,1,50])

函数传递数组时出错。由于 curve_fit 需要拟合数据，我不知道如何解决这个问题。错误如下所示:

TypeError: only size-1 arrays can be converted to Python scalars

Answer 1

用scipy.special.erfc替换math.erfc

math.erfc 适用于标量。

后者也在 array_like 上