I am trying specific rows of the array A based on the list J. For instance, it should print 1st and 4th rows of A since J=[[1,4]] and append as shown in the expected output. I also present the current output.
import numpy as np
A=np.array([[1,2,3,4,5],
[6,7,8,9,10],
[11,12,13,14,15],
[16,17,18,19,20],
[21,22,23,24,25]])
J=[[1, 4]]
for i in J[0]:
A=A[i]
print([A])
The current output is
[array([ 6, 7, 8, 9, 10])]
[array(10)]
The expected output is
[array([[ 6, 7, 8, 9, 10],
[21,22,23,24,25]])]