Суммируйте числа в списке, но меняйте их знак после появления нуля

9

Мне нужно сложить все числа в списке. Если происходит 0, начинайте вычитать, пока не начнете добавлять 0.

Например:

[1, 2, 0, 3, 0, 4] -> 1 + 2 - 3 + 4 = 4
[0, 2, 1, 0, 1, 0, 2] -> -2 - 1 + 1 - 2 = -4
[1, 2] -> 1 + 2 = 3
[4, 0, 2, 3] = 4 - 2 - 3 = -1

Вот что я попробовал:

sss = 0

for num in numbers:
    if 0 == num:
        sss = -num
    else:
        sss += num
return sss
Code34
источник

Ответы:

17

Измените знак, когда элемент списка равен 0.

result = 0
current_sign = 1
for element in your_list:
    if element == 0:
       current_sign *= -1
    result += current_sign*element
pawols
источник
2

Вот решение, которое переключается между двумя операторами (сложение и вычитание) всякий раз, когда значение в списке равно нулю:

from operator import add, sub
from itertools import cycle

cycler = cycle([add, sub])
current_operator = next(cycler)

result = 0
my_list = [1, 2, 0, 3, 0, 4]

for number in my_list:
    if number == 0:
        current_op = next(cycler)
    else:
        result = current_operator(result, number)
jfaccioni
источник
1

Попробуй это:

d = [1, 2, 0, 3, 0, 4]

sum = 0
sign = False
for i in d:
    if i == 0:
        if sign == False:
            sign = True
        else:
            sign = False
    else:
        if sign == False:
            sum += i
        else:
            sum -= i
print(sum)
Костас Харитидис
источник
Вместо вашего условия if if i == 0:вы можете просто использовать sign = not sign. Смотрите repl.it/repls/RigidCrazyDeletions
Jab
1
Также не перезаписывайте встроенную sumфункцию !! Я предполагаю , что это почему О.П. , используемые sssвместоsum
Jab
1

Еще один вариант с operatorмодулем и побитовым отрицанием ~:

import operator

def accum_on_zero(lst):
    res = 0
    ops, pos = (operator.add, operator.sub), 0
    for i in lst:
        if i == 0:
            pos = ~pos
        res = ops[pos](res, i)
    return res


print(accum_on_zero([1, 2, 0, 3, 0, 4]))     # 4
print(accum_on_zero([0, 2, 1, 0, 1, 0, 2]))  # -4 
RomanPerekhrest
источник
0
lst = [1, 2, 0, 3, 0, 4]
print(sum([v * (-1) ** lst[0:i].count(0) for i, v in enumerate(lst)]))
Кристиан Мокану
источник