如何从错误代码中删除多余的If语句

def process_operation(operation_type, data):
    if operation_type == 'add':
        return data + 1
    elif operation_type == 'subtract':
        return data - 1
    elif operation_type == 'multiply':
        return data * 2
    elif operation_type == 'divide':
        return data / 2
    else:
        raise ValueError("Unknown operation type")

def add(data):
    return data + 1

def subtract(data):
    return data - 1

def multiply(data):
    return data * 2

def divide(data):
    return data / 2

operation_map = {
    'add': add,
    'subtract': subtract,
    'multiply': multiply,
    'divide': divide
}

def process_operation(operation_type, data):
    operation = operation_map.get(operation_type)
    if operation:
        return operation(data)
    else:
        raise ValueError("Unknown operation type")