結果 : square function in math python