結果 : square of number in python using function