Select numpy 数组的行子集，基于另一个数组中的行选择

Question

我有 2 个 numpy arrays 并且我想 select 根据另一个的条件子集在其中一个中的行子集

arr1 = np.array([[1, 2, 1, 5], [3, 4, 1, 6], [2, 2, 2, 7]])
arr2 = np.array([[2, 2, 1], [2, 3, 0], [2, 1, 1]])

我想 select 仅arr1中的那些行，其中arr2中的第三个元素是 1。在这种情况下， arr2看起来像这样： np.array([[2, 2, 1], [2, 1, 1]]) arr1将变为： np.array([[1, 2, 1, 5], [2, 2, 2, 7]]) 。 可以假设它们都具有相同的行数，但可以具有不同的列数。 我怎样才能做到这一点？

Answer 1

In [500]: arr2
Out[500]: 
array([[2, 2, 1],
       [2, 3, 0],
       [2, 1, 1]])

每行的第三个元素：

In [502]: arr2[:,2]
Out[502]: array([1, 0, 1])
In [503]: arr2[:,2]==1
Out[503]: array([ True, False,  True])

将此 boolean 掩码应用于 arr1 的 select 行：

In [504]: arr1[arr2[:,2]==1]
Out[504]: 
array([[1, 2, 1, 5],
       [2, 2, 2, 7]])