mask = binary_matrix[idx[:, (0)], idx[:, (1)]] == 1
return input[mask]
res1 = landed(input)
res2 = landed2(input)
np.testing.assert_allclose(res1, res2)
Быстрая работа с массивами на основе включения элементов в бинарную матрицу