a = input() if a == 1: b = input() print(b)
a = input() if a == 1: b = input() print(b)
JOHN_16или сделать число a целочисленным: a = int(input())
* цикла у вас здесь нет, есть ветвление.* input в Python3 возвращает строку - вы же сравниваете с числом. Т.е. вам надо написать a == ‘1’