>>> import numpy as np>>> a = np.array([[1,2],[3,4]])>>> b=a.tolist()>>> b[[1, 2], [3, 4]]