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

Question

定义计算函数

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

Answer 1

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))