nodejs-polars/bin/lazy/groupby.js

Polars: Blazingly fast DataFrames in Rust, Python, Node.js, R and SQL

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports._LazyGroupBy = void 0;
const polars_internal_1 = __importDefault(require("../internals/polars_internal"));
const utils_1 = require("../utils");
const dataframe_1 = require("./dataframe");
/** @ignore */
const _LazyGroupBy = (_lgb) => {
    return {
        agg(...aggs) {
            aggs = (0, utils_1.selectionToExprList)(aggs, false);
            const ret = _lgb.agg(aggs.flat());
            return (0, dataframe_1._LazyDataFrame)(ret);
        },
        head(n = 5) {
            return (0, dataframe_1._LazyDataFrame)(_lgb.head(n));
        },
        len(name) {
            const expr = name ? polars_internal_1.default.len().alias(name) : polars_internal_1.default.len();
            return (0, dataframe_1._LazyDataFrame)(_lgb.agg([expr]));
        },
        tail(n = 5) {
            return (0, dataframe_1._LazyDataFrame)(_lgb.tail(n));
        },
    };
};
exports._LazyGroupBy = _LazyGroupBy;