Сокращение кода python?

Question

[argeo05]

Задался вопросом как сократить такой код:

if '*'==j:
    if o*t==h:print('YES')
    else:print('NO')
  elif '/'==j:
    try:
      if o/t==h:print('YES')
      else:print('NO')
    except:print('NO')
  elif '-'==j:
    if o-t==h:print('YES')
    else:print('NO')
  elif '+'==j: 
    if o+t==h:print('YES') 
    else:print('NO')
  else:0/0

Вроде операции одинаковые меняется только знак, но код выходит довольно объёмным, подскажите, пожалуйста, как его сократить

Answer 1

[Владимир Куц]

Ну наглядней точно можно сделать:

try:
    if {
        '+': lambda o, t, h: o + t == h,
        '-': lambda o, t, h: o - t == h,
        '*': lambda o, t, h: o * t == h,
        '/': lambda o, t, h: o / t == h,
      }.get(j)(o, t, h):
        print('YES')
    else:
        print('NO')
except KeyError:
    print('Неизвестная функция')
except ZeroDivisionError:
    print('Деление на ноль')

Answer 2

[Андрей]

try:
  if j in ['*', '/', '-', '+']:
      d = eval(f'o{j}t')
      print('YES' if d==h else 'NO')
  else:0/0
except:print('NO')

Comments

[Владимир Куц]

За eval запинают )

[Андрей]

Владимир Куц, Та я знаю что он опасен) Мне уже тут ни раз говорили) В любом случае он тут вроде как к месту ибо туда не вставить что-то не из списка

[Владимир Куц]

Андрей, в переменных o и t можно вставить ))
Я к тому что не надо приучивать к использованию eval

[Aleksandr Shpak]

Здесь еще упростить можно имхо. Времени нет чтоб заморочиться, но на скорую руку вышло что-то вроде такого:

try: print(f'{"YES" if {1 if (eval(f"o{j}t") if j in "*/-+" else 0) else 0} else "NO"}')
except: print("NO")

Возможно есть ошибка, но это не точно :)

[Aleksandr Shpak]

Занимательный факт: на моем комменте сломалась подсветка Python-синтаксиса

Answer 3

[Эрик Микоян]

Через библиотеку operator. Составить словарь с операторами, и тогда можно сократить весь код до 2-х условии. (Учитывая возможность деления на ноль, до 2- условии и 1 try/except)