結果 : how to write a variable squared in python