Question

I have something like

m = array([[1, 2],
            [4, 5],
            [7, 8],
            [6, 2]])

and

select = array([0,1,0,0])

My target is

result = array([1, 5, 7, 6])

I tried _ix as I read at Simplfy row AND column extraction, numpy, but this did not result in what I wanted.

p.s. Please change the title of this question if you can think of a more precise one.

Answer 1

What about using python?

result = array([subarray[index] for subarray, index in zip(m, select)])