Optei por resolver da seguinte forma:
primeiro_numero = lambda: int(input("Digite o primeiro número: "))
segundo_numero = lambda: int(input("Digite o segundo número: "))
operador = lambda: input("Escolha a operação (+, -, *, /): ")
soma = lambda x, y: x + y
subtracao = lambda x, y: x - y
multiplicacao = lambda x, y: x * y
divisao = lambda x, y: x / y if y != 0 else "Erro: Divisão por zero"
operacoes = {
"+": soma,
"-": subtracao,
"*": multiplicacao,
"/": divisao
}
numero1 = primeiro_numero()
numero2 = segundo_numero()
operacao = operador()
if operacao in operacoes:
resultado = operacoes[operacao](numero1, numero2)
print("Resultado:", resultado)