Как выполнить накопительную сумму с условным выбором на основе другого столбца в Polars? ⇐ Python

[Anonymous]

У меня есть DataFrame, и я хочу получить последние голы, забитые командами в этом кластере, чтобы применить ewm_mean(), то есть создать

Код: Выделить всё

EWM_COLUMN

как в примере ниже:
Исходная таблица



Сезон
Неделя
Домой
На выезде
HomeGoals
AwayGoals
Cluster_home
Cluster_away
Cluster_pair_key




2024
27,0
teamA
teamF
3
2
4
1
1_4


2024
27,0
командаB
командаG
1
32
2
2_2


2024
27,0
teamC
teamH
1
0
5
3
3_5


2024
27,0
teamD
teamI
0
1
< td>3
1
1_3


2024
27.0
teamE
teamJ
3
0
3
4
3_4



Глядя на teamE, я хотел учитывать голы хозяев команды E, голы хозяев команды D и голы гостей команды C. Я создал столбец Cluster_pair_key_Organized, потому что думал, что он может чем-то помочь.
Новая таблица



Сезон
Неделя
Дома
На выезде
Голы дома
Голы на выезде
Cluster_home
Cluster_away
Cluster_pair_key
EWM_COLUMN




2024
27.0
командаA
командаF
3
2
4
1
1_4
3 * 0,8 + ...


2024
27.0
teamBкомандаG
1
3
2
2
2_2
1 * 0,8 + ...


2024
27,0
teamC
teamH
1
0
5
3
3_5
0 * 0,8 + ...


< td>2024
27.0
teamD
teamI
0
1
3
1
1_3< /td>
0 * 0,8 + 0,2 * 0 + ...


2024
27.0
teamE
teamJ
3
0
3
4
3_4
3 * 0,8 + 0,2 * (0 * 0,8 + 0,2 * 0)




Подробнее здесь: https://stackoverflow.com/questions/791 ... other-colu