a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
b = np.array([[False, True, False], [True, False, False], [False, False, True]]
    )
Numpy выбирает определенный индекс столбца для каждой строки, используя список индексов