結果 : how to take square root of an array in python