Numpy.where在数组列表中,搜索整个数组

时间:2018-10-16 11:53:27

标签: python arrays numpy

我有一个数组列表,正在查看特定数组的位置:

import numpy as np
c = [np.array([0, 0]), np.array([-1,  0]), np.array([ 0, -1]), np.array([0, 1]), np.array([1, 0])]
n = np.array([0,-1])

np.where(c==n)
# array([[ True, False],
       [False, False],
       [ True,  True],
       [ True, False],
       [False, False]])

我猜它是按元素比较elemnet吗?
如何仅返回 xy均为True的位置,而不必执行另外的np.where搜索[True, True]

0 个答案:

没有答案