如何创建具有多个条件的numpy数组掩码/掩码数组?

5uzkadbs  于 2023-08-05  发布在  其他
关注(0)|答案(1)|浏览(84)

我知道在Numpy中,你可以根据条件屏蔽数组。例如,如果我想为所有大于0的值创建一个掩码数组。

arr[arr>0]

字符串
但我想有两个条件的面具。直观地看,这看起来像:

arr[arr>0 and arr<1]


但是编译器弹出一个错误:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()


我知道我可以使用for循环来解决这个问题:

masked=np.array([i if(i>0 and i<1) for i in mask])


但是有没有更优雅的解决方案,使用Numpy内置的东西?

wbgh16ku

wbgh16ku1#

你想要这个:

arr[(arr > 0) & (arr < 1)]

字符串

相关问题