ray.data.grouped_data.GroupedData.aggregate

GroupedData.aggregate(*aggs: AggregateFn) → 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.