結果 : square of each element in array python