擬合3D點python

Question

我有python代碼，產生數組x，y和z的3元組列表。我想用scipy curve_fit來擬合z = f（x，y）。下面是一些非工作代碼

A = [(19,20,24), (10,40,28), (10,50,31)] 

def func(x,y,a, b): 
    return x*y*a + b 

我怎樣才能得到蟒蛇，以適應此功能可將數據在列表中A？

Answer 1

• 的第一個參數func必須是數據（x和y）。
• 參數func的其餘參數表示參數。

所以，你需要修改func一點：

def func(data, a, b): 
    return data[:,0]*data[:,1]*a + b 

---

• 的第一個參數是curve_fit功能。
• 第二個參數是獨立data（在 形成一個陣列的x和y）。
• 第三個參數是相關數據（z）。
• 第四個參數是對參數的值的推測

因此，例如（在這種情況下a 和b）：

params, pcov = optimize.curve_fit(func, A[:,:2], A[:,2], guess) 

---

import scipy.optimize as optimize 
import numpy as np 

A = np.array([(19,20,24), (10,40,28), (10,50,31)]) 

def func(data, a, b): 
    return data[:,0]*data[:,1]*a + b 

guess = (1,1) 
params, pcov = optimize.curve_fit(func, A[:,:2], A[:,2], guess) 
print(params) 
# [ 0.04919355 6.67741935]