Полярный эквивалент pandas groupby.apply(drop_duulates) ⇐ Python

[Anonymous]

Я новичок в Polars, и мне интересно, что является эквивалентом pandas groupby.apply(drop_duulates) в Polars.
Вот фрагмент кода, который мне нужно перевести:

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

import pandas as pd

GROUP = list('123231232121212321')
OPERATION = list('AAABBABAAAABBABBBA')
BATCH = list('777898897889878987')

df_input = pd.DataFrame({'GROUP':GROUP, 'OPERATION':OPERATION, 'BATCH':BATCH})
df_output = df_input.groupby('GROUP').apply(lambda x: x.drop_duplicates())

входные данные
желаемый результат
Я попробовал следующее, но оно не выводит то, что мне нужно

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

import polars as pl

GROUP = list('123231232121212321')
OPERATION = list('AAABBABAAAABBABBBA')
BATCH = list('777898897889878987')

df_input = pl.DataFrame({'GROUP':GROUP, 'OPERATION':OPERATION, 'BATCH':BATCH})
df_output = df_input.group_by('GROUP').agg(pl.all().unique())

Если я возьму только одну группу, я получу локально то, что хочу:

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

df_part = df_input.filter(pl.col('GROUP')=='2')
df_part[['OPERATION', 'BATCH']].unique()

Кто-нибудь знает, как это сделать?

Подробнее здесь: https://stackoverflow.com/questions/762 ... duplicates