Помогите пожалуйста с кодом в Пайтон (python)
Я пишу такой код, в котором человеку предлагается ввести свой возраст, а программа должна вывести фразу "Вам ... лет/год/года". И вот тут я не понимаю, как сделать так, чтобы на определенные числа писалось "лет" (Например, Вам 30 лет), на какие-то "год" (Например, Вам 21 год) и "года" (Например, Вам 44 года). Я пробовала идти очень долгим путем сначала, типа условие "if" и перечислить все числа, когда будет выводиться, например, "год", но их там очень много! Должен же быть способ это упростить... Я пришла к выводу, что слово "год" должно писаться после любого возраста, оканчивающееся на "1"; "года" - оканчивающиеся на число от 2 до 4; все остальное - "лет". Помогите, пожалуйста!
П.С. на фото тот код, который на данный момент не правильный..
def years(y):
return ['лет', 'года', 'год'][(y // 10 % 10 != 1) * ((1 <= y % 10 <= 4) + (y % 10 == 1))]
age = int(input())
print('Вам', age, years(age) + '.')