結果 : square each element in array python