首页 文章

通过测试numpy数组中的每个元素是否在2个数字之间来创建布尔数组

提问于
浏览
0

我有一个数字的numpy数组,我想创建一个相同大小和尺寸的布尔数组,说明该元素是否位于2个数字之间 . 例如:

a=np.array([[1,2,3],[4,5,6],[7,8,9]])

我知道如果我写,

print a>3

我得到一个前三个元素为“False”而其余为“True”的数组

np.array([[False,False,False],[True,True,True],[True,True,True]])

但我想得到一个布尔数组,条件是这样的

a>3 and a<8

有没有办法在没有在for循环中逐个检查每个元素的情况下执行此操作?我有2048 * 2048的数组,这需要太长时间

有人报告说,问题与另一个解决方案是使用numpy.where函数的问题相同,但是返回条件为真的元素,我的问题是返回布尔值

1 回答

  • 0

    你可以检查这样的范围:

    print (3<a)&(a<8)
    

相关问题