python用函数写简易计算器

以下是一个简单的 Python 函数计算器示例，它支持加、减、乘、除、求幂和求根：

def calculator(operator, num1, num2):
    if operator == '+':
        return num1 + num2
    elif operator == '-':
        return num1 - num2
    elif operator == '*':
        return num1 * num2
    elif operator == '/':
        return num1 / num2
    elif operator == '**':
        return num1 ** num2
    elif operator == 'sqrt':
        return num1 ** (1/num2)
    else:
        return "Invalid operator"

# 测试
print(calculator('+', 3, 5)) # 8
print(calculator('-', 3, 5)) # -2
print(calculator('*', 3, 5)) # 15
print(calculator('/', 3, 5)) # 0.6
print(calculator('**', 3, 5)) # 243
print(calculator('sqrt', 16, 2)) # 4.0

在这个示例中，我们定义了一个名为 `calculator` 的函数，它有三个参数：操作符 `operator` 和两个数字 `num1` 和 `num2`。根据操作符的不同，函数会执行不同的计算操作，并返回结果。我们使用了 if-elif 结构来检查操作符，并执行相应的计算。如果操作符无效，则返回 "Invalid operator"。最后，我们使用 print 函数调用了 `calculator` 函数来测试它的功能。