結果 : squares in python