結果 : how to input squared in python