Как в 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[, .GRP, by = .(col1, col2)]). Спасибо!


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