zh <ddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd l mZdd lmZdd lmZdd lmZdd lmZer.funcDs (=R'@$   +.((*(;(; "  4R8#ARZZ.funcgsZ%3! IIl+// =$&$7$7KK  sA AnthcHDcgc]}|j|c}Scc}wr!)r?)rAirQs r3z/ArrowExpr.from_column_indices..vs^-Tbjjm-T-TsrGrH)narwhals._arrow.seriesrr'r()rJrKrQrFrs ` @r3from_column_indiceszArrowExpr.from_column_indicesas= 7  "T##44$$  r5cJddlm}||j|jS)Nrr)r1r2)narwhals._arrow.namespacerr'r()r+rs r3__narwhals_namespace__z ArrowExpr.__narwhals_namespace__|s!< 114==  r5cyr!)r+s r3__narwhals_expr__zArrowExpr.__narwhals_expr__sr5F)returns_scalarc|rddiSiS)N_return_py_scalarFr\)r+r^s r3_reuse_series_extra_kwargsz$ArrowExpr._reuse_series_extra_kwargss0>#U+E2Er5c(|jd|S)Ncum_sumreverse _reuse_seriesr+res r3rczArrowExpr.cum_sum!!)W!==r5c(|jd|S)Nshift)nrf)r+rls r3rkzArrowExpr.shifts!!'Q!//r5c ^jJr,tjjs d}t|s Jdfd }ndfd }j |j dzj dzjjjjS)Nz[Only aggregation or literal operations are supported in grouped `over` context for PyArrow.c~td|j}|j|jddd}|j |gd}t j |j|j}|Dcgc],}|j|jj|.c}Scc}w)NF) descending nulls_lastT)strict) rr?with_row_indexsortdroppc sort_indices get_columnr= _with_nativetake)rAtokenresultsorting_indicessorder_byr+s r3rFzArrowExpr.over..funcs6q"**E2R&&u-22%Ebggugdg;<#%//"--2F2M2M"NPVW1qxx}}_'EFWWWs1B:cPt|g\}}t|jx}rd|d}t||j dj }|j j|dd}|Dcgc]}|j|c}Scc}w)Nz Column names zV appear in both expression output names and in `over` keys. This is not yet supported.F)drop_null_keysleft_right)howleft_onright_onsuffix) r set intersectionNotImplementedErrorgroup_byagg simple_selectjoinrx) rA output_namesaliasesoverlapmsgtmpalias partition_byr+s r3rFzArrowExpr.over..funcs(I$PRTV(W% g!,/<<\JJ7J(y155.c22kk,ukEII$O&b&& 5::()# ;overrG)rArrIzSequence[ArrowSeries]) r*r kindr __class__r#r$r%r&r'r()r+rrrrFs``` r3overzArrowExpr.overs~~))) t~~/B/B CoC%c* *' '' X D*~~ ++/--8"&"="=#77 11MM  r5c(|jd|S)N cum_countrdrfrhs r3rzArrowExpr.cum_counts!!+w!??r5c(|jd|S)Ncum_minrdrfrhs r3rzArrowExpr.cum_minrir5c(|jd|S)Ncum_maxrdrfrhs r3rzArrowExpr.cum_maxrir5c(|jd|S)Ncum_prodrdrfrhs r3rzArrowExpr.cum_prods!!*g!>>r5c*|jd||S)Nrank)methodrprf)r+rrps r3rzArrowExpr.ranks!!&J!OOr5)r,z'EvalSeries[ArrowDataFrame, ArrowSeries]r-intr.strr/EvalNames[ArrowDataFrame]r0zAliasNames | Noner1ztuple[int, ...]r2rrzdict[str, Any] | NonerzImplementation | NonerINone) rJ type[Self]rErrKrr.rrIr)rJrrQrrKrrIr)rIr)rIr)r^boolrIzdict[str, Any])rerrIr)rlrrIr)rz Sequence[str]rzSequence[str] | NonerIr)rrrprrIr)__name__ __module__ __qualname__r PYARROWr__annotations__r4 classmethodrLrWrZr]rarcrkrrrrrrrewm_meanr\r5r3rr sY&4&<&05 n@>>?P Hr5rr)& __future__rtypingrrrpyarrow.computecomputervrVrnarwhals._compliantr narwhals._expression_parsingr r narwhals.exceptionsr narwhals.utilsr rrtyping_extensionsrnarwhals._arrow.dataframerrYrnarwhals._compliant.typingrrrrnarwhals.typingrrrrr\r5r3rsf" .)J73)9*&885459*&+v! *K78v!r5