結果 : how to square every element of an array in python