Numpy Array From Python Array Index