Python Format Numpy Array Indexing