Numpy，过滤器行使用 1,0 值的掩码？

Question

Opencv 有时会返回一个掩码进行过滤。

给数组 A=[[1,2],[3,4],[5,6]]
和面具mask=[1,0,1]
我应该如何敷面膜才能获得[[1,2],[5,6]] ?

我试过 A[mask==1]但它说尺寸不匹配。
np.where和 np.nonzero似乎也不起作用。

编辑:

原来A[mask==1]作品，
在我遇到的真实案例中 mask.shape是 (n,1)但不是 (n,)那个额外的 1 引起了麻烦。np.squeeze解决了问题

Answer 1

像这样？

A = np.array([[1,2],[3,4],[5,6]])
mask = np.array[1,0,1])
>>> A[np.where(mask==1),:]
array([[[1, 2],
        [5, 6]]])