Python. Задача 2-6

from math import sqrt
a, op, b = input('Введите число a, операцию (+-*/) и число b: ').split()
a = int(a)
b = int(b)
if op == '+':
    print(a, op, b, '=', a + b)
elif op == '-':
    print(a, op, b, '=', a - b)
elif op == '*':
    print(a, op, b, '=', a * b)
elif op == '/':
    print(a, op, b, '=', a / b)
else:
    print('Неизвестная операция')