Fortran Reshape Function In Python Practice