После каждого 5-го элемента вставить элемент, который равен сумме этих 5-ти элементов?

Question

[Raily Krevch]

a = [1,2,23,4,5,6,7,40,9,10,100,12,13,14,15,700,17,18,19,20]

for i in range(0, len(a), 5):
    print(sum(a[i:i+5]))

a = [1,2,23,4,5,6,7,40,9,10,100,12,13,14,15,700,17,18,19,20]
i = 5
while i < len(a):
    a.insert(i, 33333335)
    i += 6
print (a)

Я нашла сумму и поняла как вставлять, но сумма же после каждых 5-ти будет разная(не везде 35..), каким образом это можно сделать? Чтоб после каждых 5-ти выводило их сумму.

Comments

[Алан Гибизов]

А надо именно вставлять? Почему бы не разбить список по 5 элементов, к каждому кусочку добавить его сумму шестым элементом и собрать обратно в список.

Answer 1

[Andrey Dugin]

def generate(sequence, group_by=5, func=sum):
    for group in zip(*[iter(sequence)]*group_by):
        yield from group
        yield func(group)
        
print(*generate(a))

Comments

[Алан Гибизов]

Когда-то в детстве я так же ждал журналов «Техника-Молодежи» и «Наука и жизнь», как теперь жду Ваших ответов.

Ну расскажите же, что это за конструкция?
*[iter(sequence)]*group_by
Ну, первая звёздочка - понятно, разворачиваем последовательность… а вот вторая??? Умножение на group_by?

P.s.

Всякий раз полон радостного удивления: «А ещё Дугин и так может!» :)))

[Drill]

Алан Гибизов,
Допустим group_by = 3
1.

[iter(sequence)]*group_by  --> [iter(sequence), iter(sequence), iter(sequence)]

2.

zip(* [iter(sequence), iter(sequence), iter(sequence)])  --> zip(iter(sequence), iter(sequence), iter(sequence))

[o5a]

Drill, только не совсем то же самое. В данном случае получается не

zip(iter(sequence), iter(sequence), iter(sequence))

а

it = iter(sequence)
zip(it, it, it)

т.е. внутри это один и тот же итератор, за счет этого результат не дублируется, а продолжается, группируясь по group_by.

[Алан Гибизов]

Drill, спасибо огромное! Блин, но как же это всё не очевидно… хотя, если постоянно так писать, оно конечно становится как родное, «на кончиках пальцев».

[Drill]

o5a,
я все правильно написал. НЕ создается никакой переменной it

1. Просто создается три одинаковых ссылки на один и тот же объект - итератор
[iter(sequence)]*group_by == [iter(sequence), iter(sequence), iter(sequence)]

2. Далее список распаковывается и вызывается zip с аргументами. Аргументами в данном случае является одна и та же ссылка на на один и тот же объект - итератор.

3.

за счет этого результат не дублируется, а продолжается,

Поскольку имеем три одинаковых ссылки на один и тот же итератор, то и данные, само собой, извлекаются zip-oм из одного и того же итератора.
А итератор возвращает следующий элемент (по одному за раз), если он есть, или возвращает исключение, если элементы закончились.

Если я не прав, киньте в меня ссылкой на источник, где утверждается, что при распаковке создается ПЕРЕМЕННАЯ и ей присваивается ссылка

[o5a]

Drill, я только указал на разницу в том, что используется именно один и тот же объект (переменная была только для того, чтобы это показать).
Потому что в записи

[iter(sequence), iter(sequence), iter(sequence)]
или
zip(iter(sequence), iter(sequence), iter(sequence))

3 разных независимых итератора, а не ссылка на один и тот же

[Drill]

o5a,

Потому что в записи

[iter(sequence), iter(sequence), iter(sequence)]
или
zip(iter(sequence), iter(sequence), iter(sequence))

3 разных независимых итератора, а не ссылка на один и тот же

С какой стати???

Первое действие [iter(sequence)]*group_by как раз таки создает список со ссылками на один и тот же объект

In [5]: sequence = [1,2,3]

In [6]: [iter(sequence)] * 3
Out[6]: 
[<list_iterator at 0x2b63abce880>,
 <list_iterator at 0x2b63abce880>,
 <list_iterator at 0x2b63abce880>]

А дальше просто распаковывается список..

[o5a]

Drill, я указывал на эту изначальную Вашу запись

[iter(sequence)]*group_by  == [iter(sequence), iter(sequence), iter(sequence)]

в том плане, что они НЕ равносильны

[iter(sequence)]*group_by - копии одного и того же объекта-итератора
[iter(sequence), iter(sequence), iter(sequence)] - список разных итераторов

Собственно и всё, что я показывал.

[Drill]

o5a,
Я понял ). Тебя ввел в заблуждение оператор сравнения. Я не имел ввиду, что эти записи равны, хотел показать, что должно получиться в результате выполнения операции. Поправил.

[Andrey Dugin]

Алан Гибизов, рекомендую изучить Itertools Recipes, эта фишка оттуда)

[Алан Гибизов]

Andrey Dugin, Drill,
Попытался осмыслить-таки это дело.
У меня в голове получалось, если "на пальцах", что у нас есть как-бы таблица, в которой group_by колонок и каждая колонка есть ссылка на одну и ту же итерируемую последовательность a. И мы начинаем выдергивать из нее элементы, при этом метка при каждом выдергивании сдвигается во всех колонках на следующую строку (т.к. это все колонки суть одно и то же), и получается, что следующий элемент выдергивается уже со следующей строки.

Но когда прогнал это дело в дебаггере, получается, что в group на каждом цикле оказывается сразу кортеж из 5 элементов a.
И первый yield выдергивает их из уже готового кортежа. Просто выдергивает и отдает их по-очереди в вывод функции generate. Второй yield просто суммирует этот кортеж и возвращает сумму, туда же, в вывод.

Шаманство, как в group попадает этот готовый кортеж?
Ниччего не понимаю.

P.s. наверно, рано мне это еще разбирать. Пошел дальше Лутца читать, а то меньше полкнижки только проработал и решил, что уже крутой мастер. :/
P.p.s. Как раз, кстати, там и пример с умножением списков попался, вот и практика.

[Drill]

Алан Гибизов,

Шаманство, как в group попадает этот готовый кортеж?

Например, выражение zip(a, b) создает объект-итератор, из которого при каждом обороте цикла извлекается кортеж, состоящий из двух элементов. Первый берется из списка a, второй - из b.
Т.е. с каждым проходом цикла:
1 проход - извлекается (a[0], b[0])
2 проход - извлекается (a[1], b[1])
и т.д.

[Andrey Dugin]

Алан Гибизов, элемент извлекается из итератора через next(). Тогда, если представить, что it = iter(sequence), и мы берём zip(it, it, it), итерации zip будут извлекать следующие элементы:
1-я итерация: sequence[0], sequence[1], sequence[2]
2-я итерация: sequence[3], sequence[4], sequence[5]
и т.д., потому что это будет соответствовать вызовам:
1-я итерация: next(it), next(it), next(it)
2-я итерация: next(it), next(it), next(it)

[Алан Гибизов]

Andrey Dugin, я так себе это и представлял. Получается, что, считывая в первой it первый элемент, мы смещаем счётчик и во второй it уже смотрим второй элемент. Это всё происходит в течение первой итерации. Поэтому у нас сразу образовывается кортеж в первую итерацию.

Следующую итерацию мы должны бы начать со второй строки, но поскольку у нас счётчик уехал уже на 6 строку, то с неё и начинается второй цикл.

Вроде сложилось. Спасибо!

Answer 2

[Владимир Куц]

from itertools import chain

a = [1,2,23,4,5,6,7,40,9,10,100,12,13,14,15,700,17,18,19,20]
b = list(chain(*[a[x*5:x*5+5] + [sum(a[x*5:x*5+5])] for x in range(len(a)//5)]))
print(b)

# [1, 2, 23, 4, 5, 35, 6, 7, 40, 9, 10, 72, 100, 12, 13, 14, 15, 154, 700, 17, 18, 19, 20, 774]

Answer 3

[Ivan Yakushenko]

a = [1, 2, 23, 4, 5, 6, 7, 40, 9, 10, 100, 12, 13, 14, 15, 700, 17, 18, 19, 20]

for index, group in reversed(list([(i, a[i:i+5]) for i in range(0, len(a), 5)])):
    s = sum(group)
    a.insert(index + 5, s)
    print(f'{index} {group} => {s}')

# 15 [700, 17, 18, 19, 20] => 774
# 10 [100, 12, 13, 14, 15] => 154
# 5 [6, 7, 40, 9, 10] => 72
# 0 [1, 2, 23, 4, 5] => 35

# a
# [1, 2, 23, 4, 5, 35, 6, 7, 40, 9, 10, 72, 100, 12, 13, 14, 15, 154, 700, 17, 18, 19, 20, 774]

Answer 4

[Drill]

def sumelem(seq, step):
    for i, elem in enumerate(seq,1):
        if i%step:
            yield elem
        else:
            yield elem
            yield sum(seq[i-step:i])

print(*sumelem(a, 5))

Comments

[Andrey Dugin]

def sumelem(seq, step):
    for i, elem in enumerate(seq, 1):
        yield elem
        if i % step == 0:
            yield sum(seq[i-step:i])

[Drill]

Andrey Dugin,
Да, так намного лучше )

[Ivan Yakushenko]

Andrey Dugin, ты не шаришь, нужно больше елдов

[Drill]

Ivan Yakushenko,
Это к чему? Попытка под**бнуть?

[Ivan Yakushenko]

Drill, нет. Извини, если это так выглядело. Просто мой мозг иногда генерирует не всегда уместные фразы.

[Drill]

Ivan Yakushenko,
Ок, все нормально. Просто поинтересовался.
Я не гуру пайтона, так себе любитель. Написал решение с лету. Дугин подправил, так реально лучше.