}i5 ddlmZddlmZmZmZmZmZddlm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZddlmZmZmZmZddlmZmZddlm Z m!Z!m"Z"m#Z#m$Z$ersdd l%m&Z&m'Z'm(Z(dd l)m*Z*dd l+Z,dd l-Z.dd l/Z0dd l1m2Z2m3Z3m4Z4dd l5m6Z6ddl7m8Z8ddlm9Z9m:Z:m;Z;ddlm?Z?ddl@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHGdde4eedfZIedddZJgdZKGddedeeeedeeeZLGddeLeeeZMGd d!e ee!eeeefZNGd"d#eNeefeeefZOGd$d%eNeefe eeeefZPGd&d'eNeefe eeeefZQGd(d)eNeefe eeeefZRGd*d+eNeefe eeeefZSGd,d-eNeefe eeeefZTGd.d/eeeJfZUd S)0) annotations) TYPE_CHECKINGAnyGenericLiteralProtocol) CatNamespaceDateTimeNamespace ListNamespaceStringNamespaceStructNamespace)CompliantColumn)CompliantSeriesT_coEagerDataFrameAny EagerSeriesTEagerSeriesT_co NativeSeriesTNativeSeriesT_co) FromIterable FromNativeNumpyConvertible ToNarwhals)TypeVar assert_never)_StoresCompliant _StoresNativeis_compliant_seriesis_sized_multi_index_selectorunstable)IterableIteratorSequence) ModuleTypeN) NotRequiredSelf TypedDict)CompliantDataFrame)EagerNamespace)ImplementationVersion_LimitedContext)DType)Series) Into1DArray IntoDTypeMultiIndexSelector PythonLiteralRollingInterpolationMethodSizedMultiIndexSelector_1DArray _SliceIndexc$eZdZUded<ded<dS)HistDataz/NotRequired[list[float] | _1DArray | list[Any]] breakpointz2NativeSeriesT | _1DArray | _CountsT_co | list[Any]countN)__name__ __module__ __qualname____annotations__N/home/jrussi/.local/lib/python3.11/site-packages/narwhals/_compliant/series.pyr7r78s*CCCCAAAAAAr?r7 _CountsT_co Iterable[Any]T)bound covariant) CompliantSeries EagerSeriesEagerSeriesCatNamespaceEagerSeriesDateTimeNamespaceEagerSeriesHistEagerSeriesListNamespaceEagerSeriesNamespaceEagerSeriesStringNamespaceEagerSeriesStructNamespaceceZdZUded<eddZddZdd Zedd Z ddZ ddZ ddZ eddZ eddZddZdd!Zdd$Zdd&Zdd(Zedd*Zed+d,d-dd1Zdd2Zdd3Zdd4Zdd5Zdd6Zdd7Zdd:Zdd;Zdd<Zdd=Zdd>Z dd@Z!ddAZ"ddBZ#ddEZ$ddFZ%ddGZ&ddIZ'ddLZ(ddNZ)ddOZ*ddPZ+ddRZ,ddSZ-ddTZ.ddUZ/ddVZ0ddZZ1dd^Z2ddaZ3ddbZ4ddcZ5ddeZ6ddgZ7ddhZ8ddiZ9ddkZ:ddoZ;ddpZddvZ?dwdxddzZ@ddZAddZBddZCeDddZEeDddZFd,S)rEr)_implementationreturnrcdSNr>selfs r@nativezCompliantSeries.nativeWs'*sr?r%c|SrRr>rSs r@__narwhals_series__z#CompliantSeries.__narwhals_series__Ys r?r#cdSrRr>rSs r@__native_namespace__z$CompliantSeries.__native_namespace__\r?datacontextr+cdSrRr>clsr[r\s r@ from_nativezCompliantSeries.from_native]sWZWZr?Series[NativeSeriesT]c:|j|dS)Nfull)level)_versionseriesrSs r@ to_narwhalszCompliantSeries.to_narwhals_s}##D#777r?rfrcdSrRr>rTrfs r@ _with_nativezCompliantSeries._with_nativebrZr?versionr*cdSrRr>)rTrks r@ _with_versionzCompliantSeries._with_versioncrZr?r,cdSrRr>rSs r@dtypezCompliantSeries.dtypefs!cr?strcdSrRr>rSs r@namezCompliantSeries.namehs3r?rocopy bool | Noner4cdSrRr>)rTrorss r@ __array__zCompliantSeries.__array__jrZr?otherboolcdSrRr>rTrws r@ __contains__zCompliantSeries.__contains__krZr?itemMultiIndexSelector[Self]cdSrRr>rTr|s r@ __getitem__zCompliantSeries.__getitem__lrZr? Iterator[Any]cdSrRr>rSs r@__iter__zCompliantSeries.__iter__mrZr?intc*t|jSrR)lenrUrSs r@__len__zCompliantSeries.__len__ns4;r?r.cdSrRr>r^s r@ from_numpyzCompliantSeries.from_numpyqsTWTWr?N)rrrorBrrIntoDType | NonecdSrRr>)r_r[r\rrros r@ from_iterablezCompliantSeries.from_iterabless sr?cdSrRr>rzs r@__radd__zCompliantSeries.__radd__}rZr?cdSrRr>rzs r@__rand__zCompliantSeries.__rand__~rZr?cdSrRr>rzs r@__rmul__zCompliantSeries.__rmul__rZr?cdSrRr>rzs r@__ror__zCompliantSeries.__ror__rZr?cdSrRr>rSs r@allzCompliantSeries.allrZr?cdSrRr>rSs r@anyzCompliantSeries.anyrZr? ignore_nullsr1cdSrRr>)rTrs r@ any_valuezCompliantSeries.any_valuerZr?cdSrRr>rSs r@arg_maxzCompliantSeries.arg_maxrZr?cdSrRr>rSs r@arg_minzCompliantSeries.arg_minrZr?cdSrRr>rSs r@arg_truezCompliantSeries.arg_truerZr?cdSrRr>rSs r@r9zCompliantSeries.countrZr? predicatecdSrRr>)rTrs r@filterzCompliantSeries.filterrZr?cdSrRr>rSs r@firstzCompliantSeries.firstrZr?cdSrRr>rSs r@lastzCompliantSeries.lastrZr?noffsetcdSrRr>)rTrrs r@ gather_everyzCompliantSeries.gather_everyrZr?cdSrRr>rTrs r@headzCompliantSeries.headrZr?c2|dkSNr)rrSs r@is_emptyzCompliantSeries.is_emptysxxzzQr? descendingcdSrRr>)rTrs r@ is_sortedzCompliantSeries.is_sortedrZr?index int | NonecdSrRr>)rTrs r@r|zCompliantSeries.itemrZr? float | NonecdSrRr>rSs r@kurtosiszCompliantSeries.kurtosisrZr?cdSrRr>rSs r@rzCompliantSeries.lenrZr?cdSrRr>rSs r@maxzCompliantSeries.maxrZr?floatcdSrRr>rSs r@meanzCompliantSeries.meanrZr?cdSrRr>rSs r@medianzCompliantSeries.medianrZr?cdSrRr>rSs r@minzCompliantSeries.minrZr?cdSrRr>rSs r@n_uniquezCompliantSeries.n_uniquerZr?cdSrRr>rSs r@ null_countzCompliantSeries.null_countrZr?quantile interpolationr2cdSrRr>)rTrrs r@rzCompliantSeries.quantilesr?fractionwith_replacementseedcdSrRr>)rTrrrrs r@samplezCompliantSeries.samples sr?indicesvaluescdSrRr>)rTrrs r@scatterzCompliantSeries.scatterrZr?cdSrRr>rs r@shiftzCompliantSeries.shiftrZr?cdSrRr>rSs r@skewzCompliantSeries.skewrZr? nulls_lastcdSrRr>)rTrrs r@sortzCompliantSeries.sortrZr?ddofcdSrRr>rTrs r@stdzCompliantSeries.stdrZr?cdSrRr>rSs r@sumzCompliantSeries.sumrZr?cdSrRr>rs r@tailzCompliantSeries.tailrZr? pa.Array[Any]cdSrRr>rSs r@to_arrowzCompliantSeries.to_arrowrZr? separator drop_first'CompliantDataFrame[Self, Any, Any, Any]cdSrRr>)rTrrs r@ to_dummieszCompliantSeries.to_dummies25#r?cdSrRr>rSs r@to_framezCompliantSeries.to_framerZr? list[Any]cdSrRr>rSs r@to_listzCompliantSeries.to_listrZr?pd.Series[Any]cdSrRr>rSs r@ to_pandaszCompliantSeries.to_pandasrZr? pl.SeriescdSrRr>rSs r@ to_polarszCompliantSeries.to_polarsrZr?F)maintain_orderrcdSrRr>)rTrs r@uniquezCompliantSeries.uniquerZr?rparallel str | None normalizecdSrRr>)rTrrrrrs r@ value_countszCompliantSeries.value_countsrr?cdSrRr>rs r@varzCompliantSeries.varrZr?maskcdSrRr>)rTrrws r@zip_withzCompliantSeries.zip_withrZr?bins list[float]include_breakpointcdS)z(`Series.hist(bins=..., bin_count=None)`.Nr>)rTrr s r@hist_from_binszCompliantSeries.hist_from_bins r? bin_countcdS)z(`Series.hist(bins=None, bin_count=...)`.Nr>)rTrr s r@hist_from_bin_countz#CompliantSeries.hist_from_bin_countr r?rPr)rPr%)rPr#)r[rr\r+rPr%)rPra)rfrrPr%)rkr*rPr%)rPr,)rPrp)rorrsrtrPr4)rwrrPrx)r|r}rPr)rPr)rPr)r[r.r\r+rPr%) r[rBr\r+rrrprorrPr%)rwrrPr%rPrx)rrxrPr1)rrrPr%)rPr1)rrrrrPr%)rrrPr%)rrxrPrx)rrrPr)rPr)rPr)rPr)rrrr2rPr) rrrrrrxrrrPr%)rr%rr%rPr%)rrxrrxrPr%)rrrPr)rPr)rrprrxrPr)rPr)rPr)rPr)rPr)rrxrPr%) rrxrrxrrrrrxrPr)rrrwrrPr%)rr r rxrPr)rrr rxrPr)Gr:r;r<r=propertyrUrWrY classmethodr`rgrjrmrorrrvr{rrrrrrrrrrrrrrrr9rrrrrrrr|rrrrrrrrrrrrrrrrrrrrrrrrrrrrr rr>r?r@rErELs$### ***X*6555ZZZ[Z88885444::::!!!X! XJJJJ3333EEEE,,,,    WWW[W"&[0///////////....DDDD!!!!!!!!####1111))))((((<<<<'''':9991111++++    """"""""$$$$@???((((''''FFFF----'''',,,,6666GFFF''''....))))/4BBBBBB6666.---::::   X    X   r?rE)r4r.raceZdZUded<ded<ded<ded<ed,d Zed-dZd.dZddd/dZ d0dZ d1dZ d2dZ d3d Z ed4d"Zed5d$Zed6d&Zed7d(Zed8d*Zd+S)9rFr_native_seriesr)rOr*rerx _broadcastrPtuple[int, ...]c4|jSrR)rO_backend_versionrSs r@rzEagerSeries._backend_versions#44666r?rfr%Sequence[Self]cdS)aEnsure all of `series` have the same length (and index if `pandas`). Scalars get broadcasted to the full length of the longest Series. This is useful when you need to construct a full Series anyway, such as: DataFrame.select(...) It should not be used in binary operations, such as: nw.col("a") - nw.col("a").mean() because then it's more efficient to extract the right-hand-side's single element as a scalar. Nr>)r_rfs r@_align_full_broadcastz!EagerSeries._align_full_broadcasts r?valuec>||g|j|S)N)rrr\)rrr)rTrs r@ _from_scalarzEagerSeries._from_scalars!!!5' 4!HHHr?F)preserve_broadcastrr!cdS)aReturn a new `CompliantSeries`, wrapping the native `series`. In cases when operations are known to not affect whether a result should be broadcast, we can pass `preserve_broadcast=True`. Set this with care - it should only be set for unary expressions which don't change length or order, such as `.alias` or `.fill_null`. If in doubt, don't set it, you probably don't need it. Nr>)rTrfr!s r@rjzEagerSeries._with_natives  r?2EagerNamespace[Any, Self, Any, Any, NativeSeriesT]cdSrRr>rSs r@__narwhals_namespace__z"EagerSeries.__narwhals_namespace__s=@Sr?rows&SizedMultiIndexSelector[NativeSeriesT]cdSrRr>rTr&s r@_gatherzEagerSeries._gatherrZr?_SliceIndex | rangecdSrRr>r)s r@ _gather_slicezEagerSeries._gather_slicerZr?r|r}c t|ttfr||St |r||jSt|r||St|dSrR) isinstancesliceranger-rr*rUrrrs r@rzEagerSeries.__getitem__s dUEN + + ,%%d++ + t $ $ &<< ,, , *4 0 0 &<<%% %Tr?/EagerSeriesStringNamespace[Self, NativeSeriesT]cdSrRr>rSs r@rpzEagerSeries.strsFIcr?1EagerSeriesDateTimeNamespace[Self, NativeSeriesT]cdSrRr>rSs r@dtzEagerSeries.dtsGJsr?,EagerSeriesCatNamespace[Self, NativeSeriesT]cdSrRr>rSs r@catzEagerSeries.cat sCF3r?-EagerSeriesListNamespace[Self, NativeSeriesT]cdSrRr>rSs r@listzEagerSeries.list sEHSr?/EagerSeriesStructNamespace[Self, NativeSeriesT]cdSrRr>rSs r@structzEagerSeries.structsILr?NrPr)rfr%rPr)rrrPr%)rfrr!rxrPr%)rPr#)r&r'rPr%)r&r+rPr%)r|r}rPr%)rPr2)rPr4)rPr7)rPr:)rPr=)r:r;r<r=rrrrr rjr%r*r-rrpr6r9r<r?r>r?r@rFrFst#### 777X7   [ "IIIIDI       AAAAQPPPCCCCIIIXI JJJXJ FFFXF HHHXH LLLXLLLr?rFceZdZUded<eddZeddZeddZedd Zedd Z ddZ dS)_SeriesNamespacer_compliant_seriesrPc|jSrRrCrSs r@ compliantz_SeriesNamespace.compliants %%r?r)c|jjSrR)rFrOrSs r@implementationz_SeriesNamespace.implementations ~--r?rc4|jSrR)rHrrSs r@backend_versionz _SeriesNamespace.backend_version!s"33555r?r*c|jjSrR)rFrerSs r@rkz_SeriesNamespace.version%s ~&&r?rc|jjSrR)rCrUrSs r@rUz_SeriesNamespace.native)s%,,r?rfrc6|j|SrR)rFrjris r@ with_nativez_SeriesNamespace.with_native-s~**6222r?N)rPr)rPr)r@)rPr*)rPr)rfrrPr) r:r;r<r=rrFrHrJrkrUrNr>r?r@rBrBs +*** &&&X&...X.666X6'''X'---X-333333r?rBc"eZdZUded<ddZdS) rKrrCrfrPNonec||_dSrRrEris r@__init__zEagerSeriesNamespace.__init__7s!'r?N)rfrrPrP)r:r;r<r=rRr>r?r@rKrK1s8'&&&((((((r?rKceZdZdS)rGNr:r;r<r>r?r@rGrG;3r?rGceZdZdS)rHNrTr>r?r@rHrHBrUr?rHceZdZdS)rJNrTr>r?r@rJrJIrUr?rJceZdZdS)rLNrTr>r?r@rLrLPrUr?rLceZdZdS)rMNrTr>r?r@rMrMWrUr?rMceZdZUded<ded<ded<ed*d Zed+d Zd,dZddd-dZ d.dZ d/dZ d0dZ d1d"Z d2d$Zd3d&Zd4d'Zd5d(Zd)S)6rIEagerSeries[NativeSeriesT]_seriesrx _breakpoint$HistData[NativeSeriesT, _CountsT_co]_datarPrc|jjSrR)r\rUrSs r@rUzEagerSeriesHist.nativecs |""r?rfr r%cL||}||_||_|SrR)__new__r\r])r_rfr objs r@ from_serieszEagerSeriesHist.from_seriesgs(kk# , r?rcdSrRr>rSs r@rzEagerSeriesHist.to_frameprZr?both)closedstartrend num_samplesrrgLiteral['both', 'none']r4cdSrRr>)rTrhrirjrgs r@ _linear_spacezEagerSeriesHist._linear_spaceqs 3r?cdSrRr>rSs r@is_empty_serieszEagerSeriesHist.is_empty_series{rZr?c"|jrggdndgiS)N)r8r9r9)r]rSs r@ data_emptyzEagerSeriesHist.data_empty~s"262BUb2...RT Ur?argint | list[float]cdSrRr>)rTrrs r@ series_emptyzEagerSeriesHist.series_empty/2sr?rr ct|dkr||_nI|r|||_n|||_|S)N)rrqr_roru_calculate_histrTrs r@ with_binszEagerSeriesHist.with_binssi t99>>**DJJ  ! ! # # 4**400DJJ--d33DJ r?rc|dkr||_n\|r|||_n-||||_|Sr)rqr_rorury_calculate_binsrTrs r@with_bin_countzEagerSeriesHist.with_bin_countsr >>**DJJ  ! ! # # O**955DJJ--d.B.B9.M.MNNDJ r?list[float] | _1DArraycxt|tr|dd|dzn|}|ddS)Nrrx)r/rrm)rTrrrs r@_calculate_breakpointz%EagerSeriesHist._calculate_breakpoints>4>sC4H4HQt!!!Qa000cABBxr?cdSrRr>r~s r@r}zEagerSeriesHist._calculate_binsrZr?cdSrRr>rzs r@ryzEagerSeriesHist._calculate_histrvr?Nr)rfr[r rxrPr%)rPr) rhrrirrjrrgrkrPr4r)rPr^)rrrsrPr^)rr rPr%)rrrPr%)rrrsrPr)rrrPr4)rrrPr^)r:r;r<r=rrUrrdrrmrorqrur{rrr}ryr>r?r@rIrI^sA''''//// ###X#[1000+1 +***VVVV 3333?>>>333333r?rI)V __future__rtypingrrrrr!narwhals._compliant.any_namespacer r r r r narwhals._compliant.columnrnarwhals._compliant.typingrrrrrrnarwhals._translaterrrrnarwhals._typing_compatrrnarwhals._utilsrrrrrcollections.abcr r!r"typesr#pandaspdpolarsplpyarrowpatyping_extensionsr$r%r&narwhals._compliant.dataframer'narwhals._compliant.namespacer(r)r*r+narwhals.dtypesr,narwhals.seriesr-narwhals.typingr.r/r0r1r2r3r4r5r7rA__all__rErFrBrKrGrHrJrLrMrIr>r?r@rs""""""AAAAAAAAAAAAAA766666WVVVVVVVVVVV99999999B<<<<<<<<<<      >>>>>>>>>>@@@@@@<<<<<<HHHHHHHHHH%%%%%%&&&&&&                    BBBBB9gm]&BCBBB gm?dKKK    | | | | | ./}&' ] | | | ~DMDMDMDMDM/-0(=2IDMDMDMN33333()"# "2 23333>(((((_&667 O- -.(((_&667! _. ./_&667o& _. ./\#334, \+ +,\#334L! \+ +,_&667O$ _. ./B3B3B3B3B3h}k9:B3B3B3B3B3r?