python:查找子集坐标在

Question

我有一组坐标，并尝试找到坐标所在的那些子集。

import numpy as np
a=np.array([[[0,1,1],[1,1,1]],[[0,1,1],[2,1,1]],[[3,3,3],[2,2,2]]])

如果我尝试这样的事情:

print(np.argwhere(a==[[0,1,1]]))
print(a[[0,1,1]])
print(np.isin([0,1,1],a))

我得到:

[[0 0 0]
 [0 0 1]
 [0 0 2]
 [0 1 1]
 [0 1 2]
 [1 0 0]
 [1 0 1]
 [1 0 2]
 [1 1 1]
 [1 1 2]]
[[[0 1 1]
  [1 1 1]]

 [[0 1 1]
  [2 1 1]]

 [[0 1 1]
  [2 1 1]]]
[ True  True  True]

但我希望是这样的:

[true,true,false]

编辑

最好的情况是，如果我得到一个数组，其中只有属于已创建子集成员的所有其他坐标如下:

out = [[1,1,1],[2,1,1]]

Answer 1

使用 all(-1) 断言最后一个轴上的数组相等，然后使用 any(1) 检查第二个轴上是否存在这种情况:

(a == [0,1,1]).all(-1).any(1)
# array([ True,  True, False], dtype=bool)

---

关于更新:

mask = (a == [0,1,1]).all(-1)
a[mask.any(1)[:,None] & ~mask]

#array([[1, 1, 1],
#       [2, 1, 1]])