q-Ph6ddlmZddlmZmZddlmZerddlm Z m Z m Z edZ GddZ Gdd ZdS) ) annotations) TYPE_CHECKINGUnionN) DataFrame LazyFrameSeries)rrceZdZUdZded<ded<ded<ded<ded<d ed <dd ZeddZeddZddZ ddZ ddZ ddZ dS) PolarsSlicez} Apply Python slice object to Polars DataFrame or Series. Has full support for negative indexing and/or stride. intstopstartstride slice_lengthbool is_unbounded FrameOrSeriesobjreturnNonec||_dSNrselfrs S/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/_utils/slice.py__init__zPolarsSlice.__init__ lazyroriginalc|}t|tjr|n|S)z.Return lazy variant back to its original type.)collect isinstanceplr to_series)rr frames r _as_originalzPolarsSlice._as_originals6 "8R\::Quu@Q@QQrct|tjr&|n|S)z2Make lazy to ensure efficient/consistent handling.)r#r$rto_framerrs r_lazifyzPolarsSlice._lazify$s<)33 (B(BRs||~~""$$$ Rrcr||j|j|jS)z&Logic for slices with positive stride.)slicer r gather_everyrrs r_slice_positivezPolarsSlice._slice_positive)s-yyT%677DDT[QQQrct|j}||jdz|j}|dkr||n|S)z&Logic for slices with negative stride.)absrr,r rreverser-)rrr lazyslices r_slice_negativezPolarsSlice._slice_negative.sZT[!!IIdi!mT->??GGII 281**y%%f---9Lrsr,ct|j}t|j|j|j|\}}}|dkr|dko||k|_n|dko||dz k|_|jrd|_ nF|jr||_ n7||ks|dkr||ks |dkr||krdnt||z |_ |||c|_|_|_ dS)zENormalise slice bounds, identify unbounded and/or zero-length slices.r0rN) lenrr,r r stepindicesris_emptyrr1r)rr5obj_lenr r rs r _slice_setupzPolarsSlice._slice_setup4sdh--#AGQVQV<<DDWMMtV Q;;!&! B$'/D  !% G%7Q;2FD  8      !D      'D   d]] ut|| ut||  &&  .3D&* DIt{{{rc|||jdkr|jS|jrF|jdvr=|jdkr|jn|jS|jdkr;|j dkr0|jdkr%|j |j|jS|jdkr.|jdkr#|j |j dzdS| |j}|jdkr| |n| |}|||jS)zFApply a slice operation, taking advantage of any potential fast paths.r)r7r0r0)r=rrclearrrr2cloner r r,r*r.r4r')rr5lazyobjsliceds rapplyzPolarsSlice.applyQsX !   ! !8>>## #   74;'#9#9*.+//48##%%%@P@P P Z1__aDK14D4D8>>$*d.?@@ @ [1__!2a!7!78>>$)a-33 3ll48,,G;??$$W---))'22  $$VTX66 6rN)rrrr)rrr rrr)rrrr)rrrr)r5r,rr)r5r,rr) __name__ __module__ __qualname____doc____annotations__r staticmethodr'r*r.r4r=rCrrr r s IIIJJJKKKRRR\R SSS\SRRRR MMMM AAAA:777777rr c.eZdZUdZded<d dZd d Zd S) LazyPolarsSlicez Apply python slice object to Polars LazyFrame. Only slices with efficient computation paths that map directly to existing lazy methods are supported. rrrrc||_dSrrrs rrzLazyPolarsSlice.__init__wrrr5r,cP|jpd}|jpd}|jr|jdkrd}t||dkr@|dks|j3||jkr(|dcxkr|kr nn|jd}t||dkr|j ||jks.|dkrA|j:|j3|j|jcxkrdkrnn|jS|j|j|dkr|jS|dkr|j|S|dkr|jS|dkr9|jt|Sn|dcxkr|krunnr|jk|j |jdz}t|dkr|n!|t|S|dkrI|jpddkr<|j |j}|dkr|n||S|dkrQ|jJ|dkrD|j t|}|dkr|n||S|dkr]|j |jdkrK|jdn |j|z }|j ||}|dkr|n||Sd|d}t|) a Apply a slice operation. Note that LazyFrame is designed primarily for efficient computation and does not know its own length so, unlike DataFrame, certain slice patterns (such as those requiring negative stop/step) may not be supported. rr0z.negative stop is not supported for lazy slicesNz?negative stride is not supported in conjunction with start+stopr7zthe given slice zv is not supported by lazy computation Consider a more efficient approach, or construct explicitly with other methods.) r r9r ValueErrorrr?r@r-r2r1headtailr,)rr5r r9msgrrs rrCzLazyPolarsSlice.applyzsS 1v{ 6 "afqjjBCS// ! !88af&8uA$$$$$$$$$W oo% 1HH!&,!& 1HH$);!'@V@V@V@VUV@V@V@V@V@V8>>## #W_qyyx~~'''x,,T222x'')))x''))66s4yyAAAa    $     16>(--! ,,4466Ct99>>330@0@T0K0K KaZZQV[qQ..(--''C199333+;+;D+A+A AQYY16>dQhh(--E ++C199333+;+;D+A+A AQYYAFNafkk$%FN44!&5.L(.. 55C199333+;+;D+A+A A bq b b b oorN)rrrr)r5r,rr)rDrErFrGrHrrCrJrrrLrLmsYNNN]]]]]]rrL) __future__rtypingrrpolars._reexport _reexportr$polarsrrrrr rLrJrrrXs""""""''''''''13333333333/0M]7]7]7]7]7]7]7]7@jjjjjjjjjjr