I have something like
m = array([[1, 2], [4, 5], [7, 8], [6, 2]])
and
select = array([
IMHO, this is simplest variant:
m[np.arange(4), select]