Совокупная сумма в Python без текущей строки

Мне нужно найти совокупную сумму 3 предыдущих строк без расчета текущей, вот короткий пример:

ИДЕНТИФИКАТОР СУММА А 10 Б 5 Б 4 Б 1 Б 2 С 1 С 3

Вот результат, которого я хочу достичь:

ИДЕНТИФИКАТОР СУММА А 0 Б 0 Б 5 Б 9 Б 10 С 0 С 3

python pandas dataframe rolling-computation

11.12.2020 20:03

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

293

Перейти к ответу Данный вопрос помечен как решенный

Ответы 2

Можешь попробовать:

df['SUM'] = (df.groupby('ID')['SUM']
               .transform(lambda x: x.rolling(4,min_periods=1).sum())
               .sub(df['SUM'])
            )

Выход:

11.12.2020 20:08

Ответ принят как подходящий

Используйте groupby.apply с shift и rolling

df['SUM'] = df.groupby('ID')['SUM'].apply(lambda x: x.shift(fill_value=0)
                                                     .rolling(3,min_periods=1).sum())

Out[50]:
  ID   SUM
0  A   0.0
1  B   0.0
2  B   5.0
3  B   9.0
4  B  10.0
5  C   0.0
6  C   1.0

Большое спасибо!! Это работает для меня :)

— 18.12.2020 01:13

11.12.2020 20:15