使用不同的參數對公式進行迭代？

Question

我正在寫一些代碼來計算流體中兩個粒子之間的相互作用。獲得新位置的主要公式取決於其他3個函數，它們爲公式提供了必要的組件。

公式基本上是：

new position = old position + Important_matrix * Important_Force + (other force)^2 

這是我的代碼的基本佈局：

initial position of particle 1 = [1,1,1] 
initial position of particle 2 = [2,2,2] 

def GetImportantMatrix(position of first particle, position of second particle): 
'code here' 
return the_important_matrix 

def GetImportantForce(the_important_matrix): 
'code here' 
return important_force 

def GetOtherForce(): 
'code here' 
return other_force 

def getnewposition(initial position of part. 1, initial position of part. 2, the_important_matrix, important_force, other_force, iterations): 

??? 

我的實際問題是：我怎麼實現的公式功能，使其自動調用其他函數來獲取它需要的變量，而在每次迭代後更新？例如，我將如何去調用其他函數以確保當前函數獲取所需信息，並且每次都正確更新。

我見過一些這樣的例子，我在其他代碼片段中看過，但我無法弄清楚如何將它應用到我的代碼，因爲位置變量每次都會改變，這也使得其他變量每次都會改變。

我希望有道理嗎？我對Python很陌生，至今仍在學習，如果你們能指點我的任何資源來幫助我，我會很感激。

Answer 1

您的getnewposition代碼可以是這個樣子

def getnewposition(pos1, pos2): 
    importantMatrix = getImportantMatrix(pos1,pos2) 
    return pos + importantMatrix + getImportantForce(importantMatrix) + pow(getOtherForce(),2) 

然後爲你的主代碼，你需要調用是getnewposition與位置變量作爲你的論點，當您需要。

或類似的東西。我不確定你的意思，所以如果你澄清我可以幫助更多。