Как можно сократить / оптимизировать такой код на Python?

Question

[Dmitry]

Всем привет. Подскажите пожалуйста ученику, как такой код можно сократить. Спасибо.

month = int(input())
if month == 1 or month == 2 or month == 12:
    print('зима')
elif month == 3 or month == 4 or month == 5:
    print('весна')
elif month == 6 or month == 7 or month == 8:
    print('лето')
elif month == 9 or month == 10 or month == 11:
    print('осень')
else:
    print('error')

Answer 1

[Дмитрий]

использование словаря поможет значительно сократить код:

month = int(input())
seasons = {0:'зима', 1:'весна', 2:'лето', 3:'осень', 4:'зима'}
if 1 <= month <= 12: print(seasons[month//3])
else: print('error')

если ещё не знаком со словарями, во второй строке можно использовать аналогичный список

seasons = ['зима', 'весна', 'лето', 'осень', 'зима']

Comments

[Dmitry]

Спасибо Дмитрий. Тоже думал сделать через словарь, понимал что код будет намного короче.

[Dmitry]

Dmitry, спасибо

[Дмитрий]

Dmitry, двойное пожалуйста :) или это уже за UPD? Тогда - не стоит!

[Nurbol Suleimenov]

Код получился довольно маленьким, но ужасно выглядит в плане читаемости.

[Дмитрий]

Nurbol Suleimenov, так лучше?

seasons = {0:'зима', 1:'весна', 2:'лето', 3:'осень', 4:'зима'}
month = int(input())

if month < 1 or month > 12:
    result = 'error'
else:
    quarter = month//3
    result = seasons[quarter]

print(result)

[Nurbol Suleimenov]

Можете считать, что я придираюсь. И PEP8 и советы от Боба Мартина говорят нам: читаемость важнее функциональности. Код получился функциональным, и раньше я может так и написал.
Свой вариант я напишу под ответом Кирилл Кудрявцев, т.к. принимаю его вариант с некоторыми доработками

Answer 2

[Кирилл Кудрявцев]

Если минимум, то можно сделать так.

month = int(input())
if month in (1, 2, 12):
    print('зима')
elif month in (3, 4, 5):
    print('весна')
elif month in (6, 7, 8):
    print('лето')
elif month in (9, 10, 11):
    print('осень')
else:
    print('error')

Comments

[Dmitry]

а что такое in в вашем ответе?

[Кирилл Кудрявцев]

Dmitry, проверка на то что значение переменной month равно одному из значений.

[Dmitry]

Кирилл Кудрявцев, все понял, те это по сути аналог or / and... спасибо

[Nurbol Suleimenov]

Совсем не плохое решение. Конечно, была некоторое время назад идея, от любителей ООП, избавляться от `if`-ов, но я не такой :) . Я бы немного дополнил для красоты:

def get_season_name(month):
    if month in (1, 2, 12):
        return 'зима'
    elif month in (3, 4, 5):
        return 'весна'
    elif month in (6, 7, 8):
        return 'лето'
    elif month in (9, 10, 11):
        return 'осень'
    else:
        raise ValueError


if __name__ == '__main__':
    month = input('Введите номер месяца: ')
    try:
        month_name = get_season_name(int(month))
        print(month_name)
    except ValueError:
        print('error')

Answer 3

[Вадим Шаталов]

month = int(input())
seasons = {
    'winter': (1, 2, 12), 
    'spring': (3, 4, 5),
    'summer': (6, 7, 8),
    'autumn': (9, 10, 11)
}
result = 'error'
for name, monthes in seasons.items():
    if month in monthes:
        result = name
        break
print(result)

Comments

[Roman K]

O(n) сложность там, где можно сделать O(1)? Браво. Ждём ваш алгоритм с экспоненциальной сложностью. А лучше сразу с факториальной.

[Евгений]

Roman Kitaev, но ведь тут тоже сложность О(1), или у вас есть еще какие-нибудь времена года?)

[Roman K]

Евгений, гибкость все дела)

Answer 4

[PEGS]

Лучше поздно, чем никогда, такое код работает.
n = int(input())
if 1 <= n <= 12:
print(''.join((
('', 'Зима', 'Зима', 'Весна', 'Весна', 'Весна', 'Лето', 'Лето', 'Лето', 'Осень', 'Осень', 'Осень', 'Зима')[n]
)))
else:
print('Ошибка')