有没有一种很好的方法来检查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