ray.data.grouped_data.GroupedData.aggregate

GroupedData.aggregate(*aggs: ray.data.aggregate._aggregate.AggregateFn) → ray.data.dataset.Dataset

Implements an accumulator-based aggregation.

Parameters

aggs – Aggregations to do.

Returns

The output is an dataset of n + 1 columns where the first column is the groupby key and the second through n + 1 columns are the results of the aggregations. If groupby key is None then the key part of return is omitted.