結果 : square equation in python