选择np数组中列约束下的若干行数据

Question

我有两个房间

维度的A（60000784）

以及

尺寸的B（60000，1）

它们都存储整数。你知道吗

我想创建一个包含A行的nparray，这样B对应的行就是某个数字X或某个数字Y。我还想返回第二个包含B对应行的nparray。你知道吗

示例：

A

1 2
3 4
5 6
8 9


B

2
5
6
1

假设我想要A的行，使得B的对应行是1或5。那么返回的第一个nparray将如下所示：

3 4
8 9

第二个数组如下：

5
1

这样的事情在nparray中可能发生吗？你知道吗

先谢谢你

Answer 1

使用np.in1d创建布尔索引。你知道吗

A = np.array([[1, 2], [3, 4], [5, 6], [8, 9]])
B = np.array([2,5,6,1])

index = np.in1d(B, [1,5])

print A[index]

print B[index]