Как можно ужать код Python (if)?

Question

[Роман Романов]

Всем привет
Есть такой код
Как можно уменьшить его без ущерба?
Не понимаю как применить while (если он тут применим), чтобы код ды значительно меньше?

def pay():  # количество лет выплаты дивов за 7 последних лет. Если выплата не равно 0, то прибавляем +1
    i = 0
    if year2019.value != 0:
        i += 1
    else:
        pass

    if year2018.value != 0:
        i += 1
    else:
        pass

    if year2017.value != 0:
        i += 1
    else:
        pass

    if year2016.value != 0:
        i += 1
    else:
        pass

    if year2015.value != 0:
        i += 1
    else:
        pass

    if year2014.value != 0:
        i += 1
    else:
        pass

    if year2013.value != 0:
        i += 1
    else:
        pass
    pay_index = i  # получаем pay_index
    return pay_index


pay()

Answer 1

[Ivan Yakushenko]

years = [year2013, year2014, year2015]
for year in years:
    if year.value != 0:
        i += 1
    else:
        pass

Comments

[longclaps]

i += sum(y.value != 0 for y in [year2013, year2014, year2015])

[Kapustlo]

longclaps, Круто, только тут же надо будет сделать так (и функцию к одной строчке вообще можно будет свести:

return sum(map(lambda year: boolean(year.value), [year2013, year2014, year2015]))

[longclaps]

Kapustlo, в исходном вопросе нет return. Свой код поправил.

[Ivan Yakushenko]

longclaps, есть же

pay_index = i  # получаем pay_index
    return pay_index


pay()

[Kapustlo]

longclaps, только ты там скобки забыл в генераторе списка)

[longclaps]

Ivan Yakushenko, я ошибся - я реагировал не на исходный вопрос, а на твой ответ.

[Ivan Yakushenko]

longclaps, то такое. за комментарий всё-равно от меня спасибо - я с подобном использованием map и sum не сталкивался, приятно видеть различные варианты применения тех или иных функций.

[Kapustlo]

Ivan Yakushenko, это было реально круто, особенно, когда он потом на генератор списков заменил 'map'

Answer 2

[Алексей Guest007]

Вроде бы так...

years = [year2013, year2014, year2015, year2016, year2017, year2018, year2019]
result = sum(y.value for y in years if y.value)

Хотя если проверка только на 0, то на результат суммирования это не влияет, тогда вторая строка даже проще:
result = sum(y.value for y in years)