Как создать список с суммами элементов другого списка?

Question

[Timebird]

Здравствуйте! Вопрос в следующем.

Имеется список_1, в котором, допустим, 80 элементов. Нужно присвоить каждому из этих элементов список с суммой до индекса этого элемента.
Например.

Второму элементу присвоить список с суммой первого и второго элементов списка_1.
Третьему элементу присвоить список с суммой первого, второго и третьего элементов списка_1.
Четвертому элементу присвоить список с суммой первого, второго, третьего и четвертого элементов списка_1.
и т. д. до 80-го.

Вот тут пытаюсь это сделать, но безуспешно:

sum = 0
for elem in range(int(list_1[1]), int(list_1[-1])):
    list_with_summs[elem] = [] #сначала на каждый из элементов создаю свой список
    sum += elem #потом считаю сумму до соответствующего элемента
    list_with_summs[elem].append(sum) #соответствующему списку присваиваю соответствующую сумму
print('List with sums: ', list_with_summs)

list_1 подаётся на вход.

Comments

[Sly_tom_cat .]

У вас range в цикле задано по значениям первого и последнего элемента в списке, а не от первого до последнего.

Кроме того у вас каждый следующий элемент финального списка отличается от предыдущего ровно на величину элемента начального списка. если это учесть и использовать аккумулирующую переменную то все упростится до простого последовательного сложения элементов в цикле.

Answer 1

[Дмитрий Сурин]

Честно говоря, не совсем понял, из чего и что Вы хотели получить, но надеюсь код ниже Вам поможет.

sum = [2, 3, 2, 5, 6]
res = []
for k,v in enumerate(sum):
    if (k>0):
        res.append(res[k-1] + sum[k])
    else:
        res.append(sum[k])
print(sum)
print(res)
# Вывод
[2, 3, 2, 5, 6]
[2, 5, 7, 12, 18]

Comments

[Timebird]

подскажите, зачем тут нужна переменная v?

[Дмитрий Сурин]

Она не нужна. При итерации по массиву у нас есть ключ и значение массива. Если указать одну переменную при итерации то мы получим доступ к значению. Указав две мы получим доступ и ключу, и к значению. Подругому просто не получить индекс (ключ).

[Timebird]

Дмитрий Сурин: Вас примерно понял (с функцией enumerate не сталкивался, сейчас поразбираюсь ещё), спасибо Вам огромное!

[Дмитрий Сурин]

Пожалуйста. Желаю успехов! Пишите код)

[Oscar Django]

А можно так

s = [2, 3, 2, 5, 6]
[sum(s[:i+1]) for i in range(len(s))]

Вывод:
[2, 3, 2, 5, 6]
[2, 5, 7, 12, 18]

[Dimonchik]

Oscar Django: думаю, тут задание на знание алгоритма подсчета чисел Фибоначчи, у Вас же будет считаться заново каждый элемент, оно-то правильно с т.з. языка, но неочень с т.з. алгоритма

[Максим Мосейчук]

dimonchik2013: причем тут числа Фибоначчи?

[Dimonchik]

fshp: ну а какой еще алгоритм в N считает сумму N-1 предыдущих значений?

[Максим Мосейчук]

dimonchik2013: думаю, вас стоит заново почитать про числа Фибоначчи. Не N-1, а 2 предыдущих значения. Если алгоритм не использует мемонтзацию, то раскрыв все рекурсивный вызовы получим O(2^N).

[Dimonchik]

ну, найдите 80е число без вычисления всех 1-77го ))
как его решать/считать - мне известно
или, говорите, sum(s[:i+1]) окажется быстрее ?

[Максим Мосейчук]

dimonchik2013: dimonchik2013: "алгоритм в N считает сумму N-1 предыдущих значений?"
То, что написали вы, это Fib(N) = Fib(1) + Fib(2) + .. + Fib(N-2) + Fib(N-1)

Формула чисел Фибоначчи выглядит так Fib(N) = Fib(N-1) + Fib(N-2)
Разницу не находите?

[Максим Мосейчук]

dimonchik2013: И быстрее всего числа Фибоначчи через матрицу считаются, за O(log N)

[Dimonchik]

уфф, круто
и откуда Вы возьмете Fib(N-1) и Fib(N-2) значения?

[Dimonchik]

ну вот опять )))
Вы про значения скажите - откуда Fib(N-1) и Fib(N-2) берутся?
не из Fib(N-3) и Fib(N-4) .... Fib(1) случайно?

[Максим Мосейчук]

dimonchik2013: Из них, но это не является суммой N-1 элементов.

[Максим Мосейчук]

fshp: Простой пример. Fib(3). Следуя вашей логике это сумма 3-1 = 2 элементов. F(3) = F(2) + F(1) = (F(1) + F(0)) + F(1)
Элементов 3 получилось, а не 2

[Dimonchik]

fshp: ха ха )))) но по всему ряду таки разок пройтись нужно? ) или Вы ратуете за то, чтобы при каждом номере с нуля проходиться?

Answer 2

[Sly_tom_cat .]

на уровне логики описанной в задании, и простейшими действиями:

vals = [1,2,3,4,5]   # список со значениями 
acc = 0                  # аккумулятор
for i in range(len(vals)):   # проходим по списку
  acc + = vals[i]                    # увеличиваем аккумулятр на значение текущего элемента
  vals[i] = acc                       # запоминаем в оригинальном массиве накопленную сумму

Если еще проще то можно не мудрить со сканированием оригинального массива.

vals = [1,2,3,4,5]   # список со значениями 
acc = 0                  # аккумулятор
res = []
for v in vals:
  acc + = v
  res.append(acc)

Answer 3

[Максим Мосейчук]

import itertools
import operator

itertools.accumulate([1,2,3,4,5], operator.add)