結果 : square number function python