結果 : how to write a square root function in python