Question

Вася распределил своих учеников по четырём группам. У каждой группы есть номер (1, 2, 3 и 4) и цвет: первая группа взяла себе красный, вторая — синий, третья — жёлтый, четвертая — белый.
Дан цвет. Выведи номер группы. Обрати внимание, что «жёлтый» пишется через «ё».

Ввод Вывод
красный 1
жёлтый 3

Answer 1

print({'красный':1, 'синий':2, 'жёлтый':3, 'белый':4}[input()])