Function argmax_rows

python/pathway/stdlib/utils/filtering.py:8–17

(
    table: pw.Table, *on: pw.ColumnReference, what: pw.ColumnExpression
)

def argmax_rows(
    table: pw.Table, *on: pw.ColumnReference, what: pw.ColumnExpression
) -> pw.Table:
    filter = (
        table.groupby(*on)
        .reduce(argmax_id=pw.reducers.argmax(what))
        .with_id(pw.this.argmax_id)
        .promise_universe_is_subset_of(table)
    )
    return table.restrict(filter)


def argmin_rows(