Оценка математического выражения (python)

Question

print('Enter a mathematical expression: ') 
expression = input() 
space = expression.find(' ') 
oprand1 = expression[0 : space] 
oprand1 = int(oprand1) 
op = expression.find('+' or '*' or '-' or '/') 
oprand2 = expression[op + 1 : ] 
oprand2 = int(oprand2) 
if op == '+': 
ans = int(oprand1) + int(oprand2) 
print(ans) 

Так что скажем, что пользователь вводит 2 + 3 с пробелом между символами. Как мне получить его для печати 2 + 3 = 5? Мне нужен код для работы со всеми операциями.

Answer 1

Я бы предложил что-то в этих строках, я думаю, что у вас может возникнуть сложный синтаксический разбор значений из входного выражения.

Вы можете просто вызвать метод .split() в строке ввода, который по умолчанию разбивается на пробел '', поэтому строка '1 + 5' вернет ['1', '+', ' 5' ]. Затем вы можете распаковать эти значения в три переменные.

print('Enter a mathematical expression: ') 
expression = input() 
operand1, operator, operand2 = expression.split() 
operand1 = int(operand1) 
operand2 = int(operand2) 
if operator == '+': 
ans = operand1 + operand2 
print(ans) 
elif operator == '-': 
    ... 
elif operator == '/': 
    ... 
elif operator == '*': 
    ... 
else: 
    ... # deal with invalid input 

print("%s %s %s = %s" % (operand1, operator, operand2, ans))