使用輸入標簽輸入數學運算符

Question

我將以下代碼用於加法、減法、乘法和除法。 用戶將輸入自己選擇的運營商。 有沒有一種簡單的方法可以從用戶那里獲取輸入並顯示結果，而不是寫if op=='+': ？

num1 = int(input("Enter first number"))
num2 = int(input("Enter second number"))
op = input("Enter mathematical operator")
if op == '+':
    print("Result is ", num1+num2)
if op == '-':
    print("Result is ", num1-num2)
if op == '*':
    print("Result is ", num1*num2)
if op == '/':
    print("Result is ", num1/num2)
if op == '%':
    print("Result is ", num1%num2)
if op == '**'
    print("Result is ", num1**num2)
if op == '//':
    print("Result is ", num1//num2)

Answer 1

num1 = int(input("Enter first number"))
num2 = int(input("Enter second number"))
op = input("Enter mathematical operator")
if op in ('+', '-', '*', '/' , '//', '%', '**'):
    print("Result is", eval(f"{num1}{op}{num2}"))
else:
    print("Operator not recognized.")

eval是一個危險的 function，因為它可以執行任意 Python 代碼，但在這種情況下，我正在過濾輸入，所以我知道表達式是合法的。