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]])

Answer 2

我通过这样做得到了您正在寻找的结果：

[[0,1,1] in b for b in a]

我会试着弄清楚为什么isin不起作用。