結果 : how to write square root in python numpy