結果 : how to get square in python