python求前n项和的程序