L i/' 6ddlmZddlZddlmZddlmZddlmZm Z ddl Z ddl m ZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZdd lmZer ddl m!Z!m"Z"ddl#m$Z$m%Z%m&Z&ddlm'Z'ddl(m)Z)m*Z*Gddeeeee jVdfZ,y)) annotationsN)reduce)chain) TYPE_CHECKINGLiteralArrowDataFrame ArrowExpr)ArrowSelectorNamespace ArrowSeries)cast_to_comparable_string_types)EagerNamespace)combine_alias_output_namescombine_evaluate_output_names)Implementation)IteratorSequence)ChunkedArrayAny Incomplete ScalarAny)Version) IntoDTypeNonNestedLiteralceZdZejZeddZeddZeddZ ddZ ddZ ddZ ddZ ddZdd Zdd Zdd Zdd Zdd ZddZddZddZed dZ d!dZddZ d" d#dZy)$ArrowNamespacectSNrselfs _/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/narwhals/_arrow/namespace.py _dataframezArrowNamespace._dataframe$sctSrr r s r"_exprzArrowNamespace._expr(sr$ctSrr r s r"_serieszArrowNamespace._series,sr$c||_yr)_version)r!versions r"__init__zArrowNamespace.__init__0s  r$cd|DS)Nc3jK|]+}|jr|jdn |j-yw)rN) _broadcastnative.0ss r" z0ArrowNamespace.extract_native..6s'Jaq|| 9Js13)r!seriess r"extract_nativezArrowNamespace.extract_native3sK6JJr$cJjfdddjS)Nc^tjt|jgdgS)Nlen)namecontext)r from_iterabler:r0)dfr!s r"z$ArrowNamespace.len..;s())3ryy>*:PTUr$cdgS)Nr:r5_dfs r"r?z$ArrowNamespace.len..>sugr$evaluate_output_namesalias_output_namesr+r&r*r s`r"r:zArrowNamespace.len8s-zz #6#MM   r$c`dfd jfdddjS)Nc`tjgd}r|jS|S)Nliteral)datar;r<)rr=cast)_ arrow_seriesdtyper!values r"_lit_arrow_seriesz-ArrowNamespace.lit.._lit_arrow_seriesDs8&44W9dL#((// r$c|gSrr5)r>rPs r"r?z$ArrowNamespace.lit..Ms)"-.r$cdgS)NrIr5rAs r"r?z$ArrowNamespace.lit..Nsykr$rC)rLr returnrrF)r!rOrNrPs```@r"litzArrowNamespace.litCs0 zz ."9#MM   r$cjdfd }|jj|tt|S)NctjfdD}r d|D}ttj|gS)Nc3.K|] }|ywrr5r2er>s r"r4z>ArrowNamespace.all_horizontal..func..U?U!"?Uc3BK|]}|jdddyw)TN fill_nullr1s r"r4z>ArrowNamespace.all_horizontal..func..WsHA!++dD$7H)rr=roperatorand_r>r6exprs ignore_nullss` r"funcz+ArrowNamespace.all_horizontal..funcTs:,1,?,??Uu?U,UFHH8==&12 2r$rerDrEr<r>r rSzlist[ArrowSeries]r&_from_callablerrr!rdrcres `` r"all_horizontalzArrowNamespace.all_horizontalSs< 3 zz(("?"G95A )  r$cjdfd }|jj|tt|S)NctjfdD}r d|D}ttj|gS)Nc3.K|] }|ywrr5rXs r"r4z>ArrowNamespace.any_horizontal..func..crZr[c3BK|]}|jdddyw)FNr]r1s r"r4z>ArrowNamespace.any_horizontal..func..esIQ!++eT48Ir_)rr=rr`or_rbs` r"rez+ArrowNamespace.any_horizontal..funcbs:,1,?,??Uu?U,UFI&I8<<01 1r$rfrgrhrjs `` r"any_horizontalzArrowNamespace.any_horizontalas< 2 zz(("?"G95A )  r$cfdfd }|jj|tt|S)NctjfdD}d|D}ttj|gS)Nc3.K|] }|ywrr5r2exprr>s r"r4z>ArrowNamespace.sum_horizontal..func..qs$@$T"X$@r[c3DK|]}|jdddyw)rNstrategylimitr]r1s r"r4z>ArrowNamespace.sum_horizontal..func..rsLAakk!d$k?Ls )rr=rr`add)r>itr6rcs` r"rez+ArrowNamespace.sum_horizontal..funcps7$$$@%$@@BLLF8<<01 1r$rfrgrhr!rcres ` r"sum_horizontalzArrowNamespace.sum_horizontalos< 2 zz(("?"G95A )  r$c|jjjdfd }|jj |t t |S)NctttjfdD}|Dcgc]}|jddd}}|Dcgc]$}d|j j z &}}t tj|t tj|z gScc}wcc}w)Nc3.K|] }|ywrr5rus r"r4z?ArrowNamespace.mean_horizontal..func..s4P$T"X4Pr[rrx) tuplerr=r^is_nullrKrr`r{)r> expr_resultsr3r6non_narcint_64s` r"rez,ArrowNamespace.mean_horizontal..funcs !4!44P%4P!PQLIUVAakk!d$k?VFVArrowNamespace.min_horizontal..func.. init_seriesr6r3 native_seriesrcr!s` r"rez+ArrowNamespace.min_horizontal..funcx#()<)<ArrowNamespace.max_horizontal..func..rr[r) rrr=rrmax_element_wiser0rr;r*rs` r"rez+ArrowNamespace.max_horizontal..funcrrrfrgrhr}s`` r"max_horizontalzArrowNamespace.max_horizontalrr$c||jdk\rtj|dStj|dS)N)default)promote_optionsT)promote)_backend_versionpa concat_tables)r!dfss r"_concat_diagonalzArrowNamespace._concat_diagonals6  E )##CC CT22r$cttjd|D}ttjd|D}tj j ||S)Nc34K|]}|jywr) column_namesr2r>s r"r4z4ArrowNamespace._concat_horizontal..s(GR(Gc3<K|]}|jywr) itercolumnsrs r"r4z4ArrowNamespace._concat_horizontal..s*J2>>+;*Js)names)listrr=rrTable from_arrays)r!rrarrayss r"_concat_horizontalz!ArrowNamespace._concat_horizontalsRU(((G3(GGHu***Jc*JJKxx##F%#88r$c|dj}t|dddD]-\}}|j}||k7sd|d|d|d}t|tj|S)Nrr)startz>unable to vstack, column names don't match: - dataframe 0: z - dataframe z:  )r enumerate TypeErrorrr)r!rcols_0ir> cols_currentmsgs r"_concat_verticalzArrowNamespace._concat_verticalsQ$$s12wa0 %EAr??Lv%))/1&&'S<.<  n$ %$$r$c,tj|Sr)r from_namespacer s r" selectorszArrowNamespace.selectorss%44T::r$crdfd }jj|ttS)Nc,ttjfdD}|dj} rdnd}t  j |d i\}}t j} j|g||d|i| j}|gS)Nc3.K|] }|ywrr5rus r"r4z:ArrowNamespace.concat_str..func..s-I4d2h-Ir[rskip emit_null separator null_handlingr) rrr=r;rr7rbinary_join_element_wiser(r*) r>r6r;rr|separator_scalar concat_str compliantrcrdr!rs ` r"rez'ArrowNamespace.concat_str..funcs%---I5-IIJF!9>>D&K $C$$$f-$9B$ B &(%@%@J NBN 0N N %I ; r$rfrgrh)r!rrdrcres```` r"rzArrowNamespace.concat_strsA  &zz(("?"G95A )  r$cjdfd }jj|ttS)Ncjj}|tjfdD^}}t t j |jgd|D|jjgS)Nc3.K|] }|ywrr5rus r"r4z8ArrowNamespace.coalesce..func..s=Y4d2h=Yr[c34K|]}|jywr)r0r1s r"r4z8ArrowNamespace.coalesce..func..s5O1ahh5Orr) r(_align_full_broadcastrr=rrcoalescer0r;r*)r>alignrr6rcr!s` r"rez%ArrowNamespace.coalesce..funcsrLL66E#(%*=*==YSX=Y*Y#Z K&KK 2 2P5O5OP$)) MM r$rfrgrhr}s`` r"rzArrowNamespace.coalesces< zz(("?"G95A )  r$Nc|)tjt||jn|}t j |||Sr)rnullsr:typerif_else)r!whenthen otherwises r" _if_then_elsezArrowNamespace._if_then_elses7 7@6GBHHSY 2Y zz$i00r$)rSztype[ArrowDataFrame])rSztype[ArrowExpr])rSztype[ArrowSeries])r+rrSNone)r6rrSz%Iterator[ChunkedArrayAny | ScalarAny])rSr )rOrrNzIntoDType | NonerSr )rcr rdboolrSr )rcr rSr )rzSequence[pa.Table]rSzpa.Table)rSr )rcr rstrrdrrSr r)rrrrrzChunkedArrayAny | NonerSr)__name__ __module__ __qualname__rPYARROW_implementationpropertyr#r&r(r,r7r:rTrkrqr~rrrrrrrrrrr5r$r"rrs %,,O  K"K .K          " "3 9 %;;  ,/ ?C  : .-1 111* 1  1r$rr)- __future__rr` functoolsr itertoolsrtypingrrpyarrowrpyarrow.computecomputernarwhals._arrow.dataframer narwhals._arrow.exprr narwhals._arrow.selectorsr narwhals._arrow.seriesrnarwhals._arrow.utilsrnarwhals._compliantrnarwhals._expression_parsingrrnarwhals._utilsrcollections.abcrrnarwhals._arrow.typingrrrrnarwhals.typingrrrrr5r$r"rsl")4*<.A.+2MM';a1>; 288EVVWa1r$