如何從另一個多變量函數定義一個變量函數

Question

我想從一個多變量函數G定義一個變量g功能：

def dG(thetaf,psi,gamma) : 
    return 0.35*(cos(psi))**2*(2*sin(3*thetaf/2+2*gamma)+(1+4*sin(gamma)**2)*sin(thetaf/2)-sin(3*thetaf/2))+sin(psi)**2*sin(thetaf/2) 

g = lambda thetaf: dG(thetaf,psi,gamma) 

不幸的是，這是不工作和錯誤我得到的是：

只有長度爲1的陣列可以被轉換到Python標量

Answer 1

你必須定義一些默認值。如果你使用關鍵字參數來做到這一點，你甚至不需要定義一個單獨的函數。

from numpy import sin, cos, arange 

def dG(thetaf,psi=0.5,gamma=1) : 
    return 0.35*(cos(psi))**2*(2*sin(3*thetaf/2+2*gamma)+(1+4*sin(gamma)**2)*sin(thetaf/2)-sin(3*thetaf/2))+sin(psi)**2*sin(thetaf/2) 

thetaf = arange(10) 
print dG(thetaf) 
>>> [ 0.4902 0.1475 0.5077 1.6392 1.757 0.4624 -0.472 -0.2416 -0.2771 -1.3398] 

您實際上可以定義一個單獨的函數，但使用關鍵字的默認值是更清潔的選擇。

g = lambda tf: dG(tf, 0.5, 1) 
g(thetaf) 
array([ 0.4902, 0.1475, 0.5077, 1.6392, 1.757 , 0.4624, -0.472 , 
     -0.2416, -0.2771, -1.3398]) 

Answer 2

下一次，請將您的原始問題中的腳本包含在一個很好的格式中。它使幫助更快。

我認爲這只是一個簡單的錯誤。你可以分別得到theta和phi gamma和psi，但是你永遠不會使用它們。你的意思是用g作爲你的參數嗎？如果是這樣，那麼就應該是這個樣子

from numpy import sin, cos, arange, linspace, pi, zeros 
import scipy.optimize as opt  

def dG(thetaf, psi, gamma): 
    return 0.35*(cos(psi))**2*(2*sin(3*thetaf/2+2*gamma)+(1+4*sin(gamma)**2)*sin(thetaf/2)-sin(3*thetaf/2))+sin(psi)**2*sin(thetaf/2)  

nt = 100 
np = 100 
gamma = linspace(0, pi/2, nt) 
psi = linspace(0, pi/2, np) 
x = zeros((nt, np)) 
for i, theta in enumerate(gamma): 
    for j, phi in enumerate(psi): 
     print('i = %d, j = %d') %(i, j) 
     g = lambda thetaf: dG(thetaf,phi,theta) 
     x[i,j] = opt.brenth(g,-pi/2,pi/2)