按列从数组中选择按数组

Question

我有两个numpy数组：

array_1.shape = (10000,15)
array_2.shape = (435,)

我想从array_1中选择一个子数组，其中第一列（array_1[:,0]）等于array_2的元素之一。类似于array_1[array_1[:,0]==array_2]。你知道吗

Answer 1

因为这些是numpy数组，所以可以使用np.in1d来查找数组2中数组\u 1[：，0]的元素的索引，然后使用此结果对数组\u 1进行索引。你知道吗

inArray2 = np.in1d(array_1[:, 0], array_2)给出长度为array_1.shape[0]的布尔数组，如果array_1[i, 0]在数组2中，则索引i处的值等于True，否则为False。你知道吗

所以array_1[inArray2]给出了预期的输出。你知道吗