В чем суть то что вы хотите сделать? Чтобы было в 1 строку?
Можно так сделать:
[print('ДА' if a // 1000 + a % 10 == a % 1000 // 100 - a % 1000 % 100 // 10 else 'НЕТ') for a in [int(input())]]
input() - это ввод данных с клавиатуры, тк у вас а используется не 1 раз - то подставить в выражение input не получится.
Также, стоит ориентироваться не на то - сколько код короткий / помещается в строку, а на то, насколько код понятен другим.