結果 : python code to find the square of a number