我有一个数组列表,正在查看特定数组的位置:
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吗?
如何仅返回 x
和y
均为True
的位置,而不必执行另外的np.where
搜索[True, True]
?