arr = np.array([3, 7, 4])
cond = np.array([[False, True, True], [True, False, True]])

def multi_slice_max(bool_arr, x):
    return np.ma.array(x, mask=~bool_arr).argmax()
np.apply_along_axis(multi_slice_max, 1, cond, arr)
Как мне векторизовать этот цикл в numpy?