結果 : square of array in python