結果 : square of an array python without numpy