python 平方函数