Numpy Access Array Elements Using Indexes That Track