Numpy數組逐個元素的比較優化

Question

設a是長度爲n的numpy數組。 是否聲明 a == max（a） 計算表達式max（a） n次或只是一次？

Answer 1

它只評估max一次。你可以這樣自己通過編寫自己的功能測試：

def mymax(x): 
    print("Calling mymax.") 
    return max(x) 

然後嘗試

a == mymax(a) 

Answer 2

它計算max(a)一次，然後它在一個元素就會比較（標量）的（標量）結果，併爲結果創建一個布爾數組。