結果 : square calculation in python