Образец из каждой группы в кадре данных Polars? ⇐ Python

[Anonymous]

Я ищу функцию типа

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

df.group_by('column').agg(sample(10))

чтобы я мог взять десять или около того случайно выбранных элементов из каждой группы.
Это специально для того, чтобы я мог читать в LazyFrame и работать с небольшой выборкой каждой группы, а не со всем фреймом данных.
Обновление:
Одно из приближенных решений:

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

df = lf.group_by('column').agg(
pl.all().sample(.001)
)
df = df.explode(df.columns[1:])

Обновление 2
Это приблизительное решение аналогично выборке всего кадра данных и последующему группированию. Нехорошо.


Подробнее здесь: https://stackoverflow.com/questions/726 ... -dataframe