Помогите решить задачу в Python

Question

Помогите решить задачу в Python

Вера Сыскова Ученик (99), на голосовании 5 дней назад

Шахматист Вилли в свободное от соревнований время занимается программированием. Сейчас он захотел написать программу, которая бы добавляла к возрасту человека слово "год", "года" или "лет". Попробуйте написать такую же программу. Программа получает возраст человека (целое число, не превышающее 120) и выводит этот возраст со словом 'god', 'goda' или 'let'. Например, '21 god', '22 goda', '25 let'.

Answer 1

 a=int(input('adge=')) 
if a<21: 
    print(a,'лет') 
else: 
    print(a,('лет','год','года','года','года','лет','лет','лет','лет','лет')[a%10])

Answer 2

 def text_creator(age: int): 
    text = f'Вам {age}' 
    if age > 120: 
        return ValueError("ОШИБКА! Возраст не должен быть больше 120") 
    elif age % 10 == 1: 
        return f'{text} год.' 
    elif age % 10 in [2, 3, 4]: 
        return f'{text} года.' 
    else: 
        return f'{text} лет.'


age_list = [1, 2, 3, 4, 5, 6, 100, 20, 21, 24, 121, 122]
for age in age_list:
    print(text_creator(age))