「如果」在Python

Question

用來做磁盤陣列功能，我有這樣的功能：

def q(x,y): 
    r2 = (x/2.)**2 + (2.0*y)**2 
    if (r2 > 0.04): 
     return 0. 
    else: 
     return 1.5 

，我想叫它使用類似

from pylab import * 
dl=0.025 
X, Y = mgrid[-0.5:0.5:dl, -0.5:0.5:dl] 
g(X,Y) 

但很明顯，這給在比較錯誤。

這可以做，而不必爲X和Y做一個循環？ 因爲如果我想呼籲兩個雙打Q（X，Y），該用於循環必須重新實現該案例...

編輯：（添加到這個問題，因爲它太長了但可能有助於他人回答。）

看起來pylab.mgrid與numpy.mgrid相同。

調整爲numpy，該代碼

import numpy 
def q(x,y): 
    r2 = (x/2.)**2 + (2.0*y)**2 
    if (r2 > 0.04): 
     return 0. 
    else: 
     return 1.5 

dl=0.025 
X, Y = numpy.mgrid[-0.5:0.5:dl, -0.5:0.5:dl] 
q(X,Y) 

給出了這樣的錯誤

if (r2 > 0.04): 
ValueError: The truth value of an array with more than one element is 
ambiguous. Use a.any() or a.all() 

Answer 1

表達式r2 > 0.04的計算結果爲布爾值的NumPy數組，因此您不能在if語句中使用它。你可以讓它自動重新解釋爲數字，但：

def q(x,y): 
    r2 = (x/2.)**2 + (2.0*y)**2 
    return 1.5 * (r2 <= 0.04) 

Answer 2

您可以使用map給你的函數應用到每對夫婦的元素。