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)一次，然后它在一个元素就会比较（标量）的（标量）结果，并为结果创建一个布尔数组。