1>>> a = np.array([[1, 2, 3, 4],
2 [5, 6, 7, 8],
3 [9, 10, 11, 12],
4 [13, 14, 15, 16]])
5>>> np.choose([0, 2, 1, 3], a)
6array([ 1, 10, 7, 16]) # one from each row
7>>> np.take(a, [0, 2, 1, 3])
8array([1, 3, 2, 4]) # all from same row
9