結果 : how to square all elements in an array python