fiz com switch case, funciona legal:
primeiro_numero = float(input("Digite o primeiro número:"))
segundo_numero = float(input("Digite o segundo número:"))
operacao = input("Escolha a operação (| + | - | * | / |):")
def operacao_matematica(operador, x, y):
match operador:
case "+":
return x + y
case "-":
return x - y
case "*":
return x * y
case "/":
if y == 0:
return "Erro... Divisão por zero"
else:
return x / y
case _:
return "Operação inválida"
resultado = operacao_matematica(operacao, primeiro_numero, segundo_numero)
print(f"O resultado é: {resultado}")
Conte com o apoio da comunidade Alura na sua jornada. Abraços e bons estudos!