Lg;ddlmZmZddlmZmZ ddlmZmZm Z m Z m Z m Z m Z mZmZ ddlmZddlmZdd Zddd d efd Zd efdZdZefdZddZy#e$rZedeedddZ[wwxYw#e$rGdd ZeZYSwxYw))Table RecordBatch) Expressionfield) DeclarationExecNodeOptionsTableSourceNodeOptionsFilterNodeOptionsProjectNodeOptionsAggregateNodeOptionsOrderByNodeOptionsHashJoinNodeOptionsAsofJoinNodeOptionsz@The pyarrow installation is not built with support for 'acero' ()N)ScanNodeOptionsc4eZdZGddZGddZy)DatasetModuleStubc eZdZy)DatasetModuleStub.DatasetN__name__ __module__ __qualname__-/croot/pyarrow_1738229418212/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold/lib/python3.12/site-packages/pyarrow/acero.pyDatasetr3 rrc eZdZy)!DatasetModuleStub.InMemoryDatasetNrrrrInMemoryDatasetr 6rrr!N)rrrrr!rrrrr2s    rrTFc ztdt|||}|jjDcgc] }t |}}tj |tdt |g}|jjd}|*tj |tdt|g}|Scc}w)Nscan use_threadsrequire_sequenced_outputprojectfilter) rrschemanamesr from_sequencer _scan_optionsgetr )datasetr%r&declf projections filter_exprs r_dataset_to_declr3;s v[!9 ; ?HCS&&32J+J#**3/ #<s34#6""$$Z"))#. 0A0A/?1A2 ++7P0P  C*=$=00;&C S,?%?//<'C#**3/""%%c*E@J! )+7JK (($ );<==[=9Le ** *!!,//122U  s) O/3O/ O4 O4c t|ttjfst dt |t|ttjfst dt |t|t tfs|g}t|t tfs|g}|jjD cgc] } | |g|zvr| } } t|jjt| z} | rtdj| t|tjrt||d} ntdt|} t|tjrt||d} ntdt|} t!|||||}td|| | g}|j#|}|tk(r|S|tj$k(rtj$|St d cc} w) a- Perform asof join of two tables or datasets. The result will be an output table with the result of the join operation Parameters ---------- left_operand : Table or Dataset The left operand for the join operation. left_on : str The left key (or keys) on which the join operation should be performed. left_by: str or list[str] The left key (or keys) on which the join operation should be performed. right_operand : Table or Dataset The right operand for the join operation. right_on : str or list[str] The right key (or keys) on which the join operation should be performed. right_by: str or list[str] The right key (or keys) on which the join operation should be performed. tolerance : int The tolerance to use for the asof join. The tolerance is interpreted in the same units as the "on" key. output_type: Table or InMemoryDataset The output type for the exec plan result. Returns ------- result_table : Table or InMemoryDataset r5zOColumns {} present in both tables. AsofJoin does not support column collisions.Tr$r8asofjoinr=r7rA)rBrrCrrDrErFrGr)r*rI ValueErrorformatr3rr rrNr!)rPleft_onleft_byrRright_onright_by tolerancer%rWr^r]columns_collisionsrbrcrdr/rks r_perform_join_asofrwsB lUBJJ$7 89$|:L9MNOO meRZZ%8 99$}:M9NOPP gt} -) h .:%++111 xj8+ + 1\00667#m:LL !!'(:!;  , +& #%)+ " 2<@ -,' {%)+ # 2=A $(HiI I{L.I D==[=9Le ** *!!,//122Ys9G>c Bd}t|trtj|g}d}t j t dt |t dt|g}|jd}|r!|jjd}|S)a}Filter rows of a table based on the provided expression. The result will be an output table with only the rows matching the provided expression. Parameters ---------- table : Table or RecordBatch Table that should be filtered. expression : Expression The expression on which rows should be filtered. Returns ------- Table FTr8)r>r(r7r) rBrr from_batchesrr+r r rNcombine_chunks to_batches)table expressionis_batchr/results r _filter_tabler^s"H%%""E7+  $ $N,B5,IJH&7 &CD& D]]t] ,F&&(335a8 Mrc vt|tjrt|d}nt dt |}t dt |fi|}t j||g}|jd}|tk(r|S|tjk(rtj|Std)NTr7r8order_byrA) rBrCrr3rr r r+rNrr!rD)table_or_dataset sort_keysrWkwargs data_sourcerr/rks r _sort_sourcer~s"BJJ/&'7TJ ! 23CD :'9)'Nv'NOH  $ $k8%< =D==T=2Le ** *!!,//122rc tjtdt|tdt||g}|j |S)Nr8 aggregate)keysr7)rr+r r rN)r| aggregatesrr%r/s r _group_byrsL  $ $N$:5$ABK!5jt!LM& D ==[= 11r)TF)T) pyarrow.librrpyarrow.computerrpyarrow._acerorrr r r r r rr ImportErrorexcstrpyarrow.datasetr.rCpyarrow._datasetrrr3rlrwrrrrrrrs2+-   "  02#"%# n3f/3#(X3v@;@3,2Y   J3s8*TUV   Bs(A A5A2A--A25B B