Pulling elements from 2D array based on index array

Question

I have an array as:

import numpy as np
A = np.arange(15).reshape(3, 5)

I also have an index array as:

ind = np.asarray([1,2,0,2,2])

The elements of ind represent the row number of A for each column of A.

i.e

I want to pull ind[0] = 1 element from column 0 of A I want to pull ind[4] = 2 element from column 4 of A

Desired output is:

5, 11, 2, 13, 14

Divakar · Accepted Answer · 2016-10-06 15:58:18Z

2

A[ind,np.arange(ind.size)]

answered Oct 6, 2016 at 15:58

Divakar

222k19 gold badges273 silver badges374 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1