python求平方的代码