python编写计算函数

下面是一个简单的Python计算函数的示例，可以进行加、减、乘、除四种基本运算：

def calculate(num1, num2, operator):
    if operator == '+':
        return num1 + num2
    elif operator == '-':
        return num1 - num2
    elif operator == '*':
        return num1 * num2
    elif operator == '/':
        if num2 == 0:
            return "Error: division by zero"
        else:
            return num1 / num2
    else:
        return "Error: invalid operator"

这个函数接受三个参数：两个数字和一个运算符。根据运算符不同，函数会执行加、减、乘、除运算，或者返回错误信息。

例如，如果要计算3加5，可以调用函数如下：

result = calculate(3, 5, '+')
print(result)

这会输出8。同样的，如果要计算10除以0，会返回错误信息：

result = calculate(10, 0, '/')
print(result)

这会输出"Error: division by zero"。