Python Numpy通过搜索2个值来查找由5个值组成的数组的行索引

Question

如果这个标题不是新名词的话。在

假设我们有一个numpy数组作为世界地图。 参数是（x y r g b）-都是int16

示例：

a = np.array([[  0,   0,   0, 255,   0], #index 0
              [  0,   1,   0,   0, 255], #index 1
              [  0,   2,   0, 255,   0]]) #index 2

现在我们要找到x和y值（0，2）的行的索引值，因此是索引为2的行。在

^{pr2}$

在不输入其他值（rgb）的情况下，我如何做到这一点？基本上，我们正在搜索一个包含两个值的五值行-我该怎么做？在

Answer 1

您可以将行切片到第二列，并检查它们是否等于[0,2]。然后使用^{}设置axis到{}将满足所有条件的设置为True，并使用布尔数组索引ndarray：

a = np.array([[  0,   0,   0, 255,   0],
              [  0,   1,   0,   0, 255],
              [  0,   2,   0, 255,   0]])

a[(a[:,:2] == [0,2]).all(1)]
# array([[  0,   2,   0, 255,   0]])