結果 : how to write squared in python