Как в Polars создать счетчик группы/идентификатор группы? ⇐ Python

[Anonymous]

Как получить такой столбец group_id, группируя его по столбцам col1 и col2 ?



col1
col2
group_id




A
Z
1


А
Да
< td>2


A
Z
1


BZ
3



на основе такого DataFrame:

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

df = pl.DataFrame({
'col1': ['A', 'A', 'A', 'B'],
'col2': ['Z', 'Y', 'Z', 'Z']}
)

Другими словами, я ищу поляры, эквивалентные R data.table .GRP (df[, group_id:=.GRP, by = .(col1, col2)]). Спасибо!
Контекст. Я хочу создать идентификатор события, поскольку в моих данных есть много подробных строк для одного события. Как только идентификатор события будет создан, я буду использовать его для выполнения различных оконных операций. Я предпочитаю иметь этот идентификатор события, а не хранить список группирующих переменных.

Подробнее здесь: https://stackoverflow.com/questions/792 ... r-group-id