結果 : how to square every number in an array python