python一道关于异常处理的问题

题目

math模块包含了大量用于处理数值相关计算的函数和常量。但是它并不能识别复数，创建一个safe_sqrt()函数，它封装math.sqrt()并能处理数值，返回一个对应的复数。（复数的处理可以用cmath模块）

在这里要用到try-except语句。

import math
import cmath

def safe_sqrt(n):
    try:
        value = math.sqrt(n)
    except ValueError:
       value = cmath.sqrt(n)
    print(value)
n = input('请输入你要计算的数值: ')
safe_sqrt(int(n))

请输入你要计算的数值: 2
1.4142135623730951
请输入你要计算的数值: -1
1j