定义一个用于计算的函数,请按照以下要求完成。

定义计算函数

1)函数有3个参数,operation可以接收“+”、“-”、“*”和“/”符号;num1和num2分别用于接收整数。
2)使用if-elif语句判断,如果为“+”,返回num1和num2的和;如果为“-”,返回num1和num2的差,以此类推。
3)注意,处理num2不为0的情况。

def calculation(operation, num1, num2):
    if num2 != 0:
        if operation == '+':
            return num1 + num2
        elif operation == '-':
            return num1 - num2
        elif operation == '*':
            return num1 * num2
        elif operation == '/':
            return num1 / num2
    else:
        return 'num2不能为0'


if __name__ == '__main__':
    print(calculation('/', 1, 3))