我们可以在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图书馆可能吗?你知道吗
由于比较是按元素进行的,因此需要在第一个轴上使用
all
来减少比较:相关问题 更多 >
编程相关推荐