結果 : square all elements in array python