結果 : how to square each element in an array python