根据矢量型元素获取numpy数组掩码

2024-03-29 15:35:08 发布

您现在位置:Python中文网/ 问答频道 /正文

我们可以在numpy数组中找到标量的索引,如下所示:

import numpy as np
array = np.array([1,2,3])
mask = (array == 2) #mask is now [False,True,False]

当元素是向量时:

import numpy as np
array = np.array([[1,2],[1,4],[5,6]])
mask = (array == [1,4]) #mask is now [[True,False],[True,True],[False,False]]

实际上,我想生成一个类似于第二个示例中的第一个代码段的掩码。你知道吗

mask = [False,True,False]

这在numpy图书馆可能吗?你知道吗


Tags: importnumpyfalsetrue元素示例isas
1条回答
网友
1楼 · 发布于 2024-03-29 15:35:08

由于比较是按元素进行的,因此需要在第一个轴上使用all来减少比较:

(array == [1, 4]).all(axis=1)
Out: array([False,  True, False], dtype=bool)

相关问题 更多 >