Select Range In Array Python Size Command In Matlab