小编dca*_*ida的帖子

我想知道如何np.where与2D阵列一起使用

我有以下数组：

arr1 = np.array([[ 3.,  0.],
                 [ 3.,  1.],
                 [ 3.,  2.],
                 [ 3.,  3.],
                 [ 3.,  6.],
                 [ 3.,  5.]])

我想找到这个数组：

arr2 = np.array([3.,0.])

但是当我使用时np.where()：

np.where(arr1 == arr2)

它返回：

(array([0, 0, 1, 2, 3, 4, 5]), array([0, 1, 0, 0, 0, 0, 0]))

我不明白这是什么意思。有人可以帮我解释一下吗？