結果 : how to square an array in python