結果 : how to square root an array in python