結果 : how to write code for square root in python