有沒有一種很好的方法來檢查numpy數組元素是否在一個範圍內？

Question

我想寫：

assert np.all(0 < a < 2) 

其中a是numpy陣列，但它不工作。什麼是寫這個的好方法？

Answer 1

你可以使用numpy.logical_and：

>>> a = np.repeat(1, 10) 
>>> np.logical_and(a > 0, a < 2).all() 
True 

或使用&。

>>> ((0 < a) & (a < 2)).all() 
True