Indexing Into Numpy Array Python Code Generator