結果 : square elements in array python