I have an array A with shape [2, 512, 21128] and another list [7,41]
I want to create a new array with shape [2,21128] from A[0,7,:] and A[1,41,:].
I can use a loop to do such things, is there any convenient way to do this? I try to find it online but I actually do not know how to ask this question.
You can use advanced indexing:
arr = np.random.rand(2, 512, 21128) out = arr[[0, 1], [7, 41]]