How To Index An Array Python