結果 : square in python code