Numpy multidimensional indexing

# Let's say you have a batch of 256 inputs, each one has a set of 64 points.
x.shape --> (256, 64, 3)

# Now assume you want to index each image and draw 128 samples of those points,
# with an indices matrix.
idx.shape --> (256, 128)

# Do
y = x[np.arange(x.shape[0]), idx.T].reshape((1, 0, 2))
y.shape --> (256, 128, 3)

Similar codes

0 vote193 views3 answers
How to use group by in python to get 15 mins candle data from 1 min candle
Python
19 votes187 views5 answers
Pandas dataframe column names
Python
1 vote185 views1 answer
Python list index time complexity
Python
4 votes169 views3 answers
Iterate python
Python
0 vote211 views1 answer
Generate random numbers following Poisson distribution, Geometric Distribution, Uniform Distribution, and Normal Distribution, and plot them
Python
0 vote167 views1 answer
Switch between frames in tkinter
Python
27 votes183 views6 answers
How to check if there are duplicates in a list python
Python
4 votes177 views1 answer
Python zip file open as text
Python
1 vote199 views1 answer
Discord.py message user
Python
43 votes173 views3 answers
Change the current working directory in python
Python