How To Index A 2d Array Python