L iE ddlmZddlmZmZddlmZer"ddlmZm Z ddl m Z ddl mZddlmZmZddlmZGd d ZGd d Zy )) annotations) TYPE_CHECKINGcast)is_sequence_of)IteratorSequence)GroupBy) LazyGroupBy)PolarsDataFramePolarsLazyFrame) PolarsExprc\eZdZUded<ded<ed dZ d dZd dZd dZy ) PolarsGroupByr _compliant_frame NativeGroupBy_groupedc|jSNrselfs _/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/narwhals/_polars/group_by.py compliantzPolarsGroupBy.compliant$$$c0t||_|r|j|n||_t |t r+|j jj||_ y|j jjd|D|_ y)Nc34K|]}|jywrnative.0args r z)PolarsGroupBy.__init__..&/Ks /K list_keys drop_nullsrrstrrrgroup_byrrdfkeysdrop_null_keyss r__init__zPolarsGroupBy.__init__x$Z 7E d 32dC( NN ! ! * *4 0 &&///Kd/KK rc||jjd|D}|jj|S)Nc34K|]}|jywrrr s rr#z$PolarsGroupBy.agg..*&Bcszz&Br%raggr _with_nativeraggs agg_results rr6zPolarsGroupBy.agg)1]]&&&BT&BB ~~**:66rc#K|jD]7\}}ttd||jj |f9yw)Nr*)rtuplerrr7)rkeyr-s r__iter__zPolarsGroupBy.__iter__-sE}} KGCUC()4>>+F+Fr+JJ J KsAA N)returnr )r-r r.$Sequence[PolarsExpr] | Sequence[str]r/boolr@None)r9r r@r )r@z1Iterator[tuple[tuple[str, ...], PolarsDataFrame]]) __name__ __module__ __qualname____annotations__propertyrr0r6r?rrrrsV%% %%  3     7KrrcTeZdZUded<ded<ed dZ d dZd dZy) PolarsLazyGroupByr rNativeLazyGroupByrc|jSrrrs rrzPolarsLazyGroupBy.compliant6rrc0t||_|r|j|n||_t |t r+|j jj||_ y|j jjd|D|_ y)Nc34K|]}|jywrrr s rr#z-PolarsLazyGroupBy.__init__..Gr$r%r&r,s rr0zPolarsLazyGroupBy.__init__:r1rc||jjd|D}|jj|S)Nc34K|]}|jywrrr s rr#z(PolarsLazyGroupBy.agg..Kr4r%r5r8s rr6zPolarsLazyGroupBy.aggJr;rN)r@r )r-r r.rAr/rBr@rC)r9r r@r )rDrErFrGrHrr0r6rIrrrKrK2sP%% %%  3     7rrKN) __future__rtypingrrnarwhals._utilsrcollections.abcrrpolars.dataframe.group_byr rpolars.lazyframe.group_byr rLnarwhals._polars.dataframer r narwhals._polars.exprr rrKrIrrrZs7"&*2BJK0KKB77r