結果 : how to code a square numbers in python