o Lh5 @sddlmZddlmZmZmZmZmZddlm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZddlmZmZmZmZddlmZmZddlmZm Z m!Z!m"Z"m#Z#erdd l$m%Z%m&Z&m'Z'dd l(m)Z)dd l*Z+dd l,Z-dd l.Z/dd l0m1Z1m2Z2m3Z3dd l4m5Z5ddl6m7Z7m8Z8ddl9m:Z:ddlm;Z;mm?Z?ddl@mAZAddlBmCZCmDZDmEZEmFZFmGZGmHZHmIZIGddde3eedfZJeddddZKgdZLGdddedeeeedeeeZMGdd d eMeeeZNGd!d"d"eee eeeefZOGd#d$d$eOeefeeefZPGd%d&d&eOeefe eeeefZQGd'd(d(eOeefe eeeefZRGd)d*d*eOeefe eeeefZSGd+d,d,eOeefe eeeefZTGd-d.d.eOeefe eeeefZUGd/d0d0eeeKfZVd S)1) annotations) TYPE_CHECKINGAnyGenericLiteralProtocol) CatNamespaceDateTimeNamespace ListNamespaceStringNamespaceStructNamespace)CompliantColumn)CompliantSeriesT_coEagerDataFrameAnyEagerSeriesT_co NativeSeriesTNativeSeriesT_co) FromIterable FromNativeNumpyConvertible ToNarwhals)TypeVar assert_never)_StoresCompliant _StoresNativeis_compliant_seriesis_sized_multi_index_selectorunstable)IterableIteratorSequence) ModuleTypeN) NotRequiredSelf TypedDict)CompliantDataFrame) CompliantExpr EagerExpr)EagerNamespace)ImplementationVersion_LimitedContext)DType)Series) Into1DArray IntoDTypeMultiIndexSelectorRollingInterpolationMethodSizedMultiIndexSelector_1DArray _SliceIndexc@seZdZUded<ded<dS)HistDataz/NotRequired[list[float] | _1DArray | list[Any]] breakpointz2NativeSeriesT | _1DArray | _CountsT_co | list[Any]countN)__name__ __module__ __qualname____annotations__r<r<[/var/www/html/Persson_Maskin/env/lib/python3.10/site-packages/narwhals/_compliant/series.pyr57s  r5 _CountsT_co Iterable[Any]T)bound covariant) CompliantSeries EagerSeriesEagerSeriesCatNamespaceEagerSeriesDateTimeNamespaceEagerSeriesHistEagerSeriesListNamespaceEagerSeriesNamespaceEagerSeriesStringNamespaceEagerSeriesStructNamespacec@seZdZUded<edddZeddd Zedd d ZdddZdddZ dddZ dddZ dd d!Z dd#d$Z dd&d'Zdd)d*Zdd-d.Zdd0d1Zedd5d6Zedd8d9Zed:d;d<dd@dAZddCdDZddEdFZddGdHZddIdJZddKdLZddMdNZddOdPZddQdRZddSdTZddUdVZddWdXZ ddZd[Z!dd^d_Z"dd`daZ#ddbdcZ$ddedfZ%ddidjZ&ddldmZ'ddndoZ(ddpdqZ)ddsdtZ*ddudvZ+ddwdxZ,ddydzZ-dd{d|Z.dddZ/dddZ0dddZ1dddZ2dddZ3dddZ4dddZ5dddZ6dddZ7dddZ8dddZ9dddZ:dddZ;dddZdddZ?dddZ@dddZAeBdddZCeBdddĄZDd;S)rBr)_implementationreturnr,cCdSNr<selfr<r<r=dtypeUzCompliantSeries.dtypestrcCrMrNr<rOr<r<r=nameWrRzCompliantSeries.namercCrMrNr<rOr<r<r=nativeYrRzCompliantSeries.nativer#cCs|SrNr<rOr<r<r=__narwhals_series__[rRz#CompliantSeries.__narwhals_series__r!cCrMrNr<rOr<r<r=__native_namespace__^z$CompliantSeries.__native_namespace__rQrcopy bool | Noner3cCrMrNr<)rPrQrYr<r<r= __array___rXzCompliantSeries.__array__otherboolcCrMrNr<rPr\r<r<r= __contains__`rXzCompliantSeries.__contains__itemMultiIndexSelector[Self]cCrMrNr<rPr`r<r<r= __getitem__arXzCompliantSeries.__getitem__ Iterator[Any]cCrMrNr<rOr<r<r=__iter__brXzCompliantSeries.__iter__intcCs t|jSrN)lenrUrOr<r<r=__len__c zCompliantSeries.__len__seriescCrMrNr<rPrjr<r<r= _with_nativefrXzCompliantSeries._with_nativeversionr*cCrMrNr<)rPrmr<r<r= _with_versiongrXzCompliantSeries._with_versionCompliantExpr[Any, Self]cCrMrNr<rOr<r<r=_to_exprhrXzCompliantSeries._to_exprdatacontextr+cCrMrNr<clsrqrrr<r<r= from_nativeirRzCompliantSeries.from_nativer.cCrMrNr<rsr<r<r= from_numpykrRzCompliantSeries.from_numpyN)rTrQr?rTIntoDType | NonecCrMrNr<)rtrqrrrTrQr<r<r= from_iterablems zCompliantSeries.from_iterableSeries[NativeSeriesT]cCs|jj|ddS)Nfull)level)_versionrjrOr<r<r= to_narwhalswzCompliantSeries.to_narwhalscCrMrNr<r^r<r<r=__radd__zrXzCompliantSeries.__radd__cCrMrNr<r^r<r<r=__rand__{rXzCompliantSeries.__rand__cCrMrNr<r^r<r<r=__rmul__|rXzCompliantSeries.__rmul__cCrMrNr<r^r<r<r=__ror__}rXzCompliantSeries.__ror__cCrMrNr<rOr<r<r=all~rXzCompliantSeries.allcCrMrNr<rOr<r<r=anyrXzCompliantSeries.anycCrMrNr<rOr<r<r=arg_maxrXzCompliantSeries.arg_maxcCrMrNr<rOr<r<r=arg_minrXzCompliantSeries.arg_mincCrMrNr<rOr<r<r=arg_truerXzCompliantSeries.arg_truecCrMrNr<rOr<r<r=r7rXzCompliantSeries.count predicatecCrMrNr<)rPrr<r<r=filterrXzCompliantSeries.filternoffsetcCrMrNr<)rPrrr<r<r= gather_everyrXzCompliantSeries.gather_everycCrMrNr<rPrr<r<r=headrXzCompliantSeries.headcCs |dkSNr)rgrOr<r<r=is_empty zCompliantSeries.is_empty descendingcCrMrNr<)rPrr<r<r= is_sortedrXzCompliantSeries.is_sortedindex int | NonecCrMrNr<)rPrr<r<r=r`rXzCompliantSeries.item float | NonecCrMrNr<rOr<r<r=kurtosisrXzCompliantSeries.kurtosiscCrMrNr<rOr<r<r=rgrXzCompliantSeries.lencCrMrNr<rOr<r<r=maxrXzCompliantSeries.maxfloatcCrMrNr<rOr<r<r=meanrXzCompliantSeries.meancCrMrNr<rOr<r<r=medianrXzCompliantSeries.mediancCrMrNr<rOr<r<r=minrXzCompliantSeries.mincCrMrNr<rOr<r<r=n_uniquerXzCompliantSeries.n_uniquecCrMrNr<rOr<r<r= null_countrXzCompliantSeries.null_countquantile interpolationr1cCrMrNr<)rPrrr<r<r=rzCompliantSeries.quantilefractionwith_replacementseedcCrMrNr<)rPrrrrr<r<r=samplezCompliantSeries.sampleindicesint | Sequence[int]valuescCrMrNr<)rPrrr<r<r=scatterrXzCompliantSeries.scattercCrMrNr<rr<r<r=shiftrXzCompliantSeries.shiftcCrMrNr<rOr<r<r=skewrXzCompliantSeries.skew nulls_lastcCrMrNr<)rPrrr<r<r=sortrXzCompliantSeries.sortddofcCrMrNr<rPrr<r<r=stdrXzCompliantSeries.stdcCrMrNr<rOr<r<r=sumrXzCompliantSeries.sumcCrMrNr<rr<r<r=tailrXzCompliantSeries.tail pa.Array[Any]cCrMrNr<rOr<r<r=to_arrowrXzCompliantSeries.to_arrow separator drop_first'CompliantDataFrame[Self, Any, Any, Any]cCrMrNr<)rPrrr<r<r= to_dummiesrzCompliantSeries.to_dummiescCrMrNr<rOr<r<r=to_framerXzCompliantSeries.to_frame list[Any]cCrMrNr<rOr<r<r=to_listrXzCompliantSeries.to_listpd.Series[Any]cCrMrNr<rOr<r<r= to_pandasrXzCompliantSeries.to_pandas pl.SeriescCrMrNr<rOr<r<r= to_polarsrXzCompliantSeries.to_polarsF)maintain_orderrcCrMrNr<)rPrr<r<r=uniquerXzCompliantSeries.uniquerparallel str | None normalizecCrMrNr<)rPrrrTrr<r<r= value_countsrzCompliantSeries.value_countscCrMrNr<rr<r<r=varrXzCompliantSeries.varmaskcCrMrNr<)rPrr\r<r<r=zip_withrXzCompliantSeries.zip_withbins list[float]include_breakpointcCdS)z(`Series.hist(bins=..., bin_count=None)`.Nr<)rPrrr<r<r=hist_from_binszCompliantSeries.hist_from_bins bin_countcCr)z(`Series.hist(bins=None, bin_count=...)`.Nr<)rPrrr<r<r=hist_from_bin_countrz#CompliantSeries.hist_from_bin_count)rLr,)rLrSrLr)rLr#)rLr!)rQrrYrZrLr3)r\rrLr])r`rarLr)rLrd)rLrf)rjrrLr#)rmr*rLr#)rLro)rqrrrr+rLr#)rqr.rrr+rLr#) rqr?rrr+rTrSrQrxrLr#)rLrz)r\rrLr#rLr])rrrLr#)rrfrrfrLr#)rrfrLr#)rr]rLr])rrrLr)rLr)rLr)rLr)rrrr1rLr) rrrrrr]rrrLr#)rrrrrLr#)rr]rr]rLr#)rrfrLr)rLr)rrSrr]rLr)rLr)rLr)rLr)rLr)rr]rLr#) rr]rr]rTrrr]rLr)rrr\rrLr#)rrrr]rLr)rrfrr]rLr)Er8r9r:r;propertyrQrTrUrVrWr[r_rcrerhrlrnrp classmethodrurvryr~rrrrrrrrrr7rrrrrr`rrgrrrrrrrrrrrrrrrrrrrrrrrrrrrrr<r<r<r=rBKs                                                           rB)r3r.rzc@seZdZUded<ded<ded<ded<edddZddd?ddZ d@ddZ dAdd Z dBd#d$Z dCd&d'Z dDd*d+ZedEd-d.ZedFd0d1ZedGd3d4ZedHd6d7ZedId9d:Zd;S)JrCr_native_seriesr)rKr*r}r] _broadcastrLtuple[int, ...]cC |jSrN)rK_backend_versionrOr<r<r=r zEagerSeries._backend_versionrjr#Sequence[Self]cGr)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<)rtrjr<r<r=_align_full_broadcastsz!EagerSeries._align_full_broadcastvaluecCs|j|g|j|dS)N)rTrr)ryrT)rPrr<r<r= _from_scalarszEagerSeries._from_scalarF)preserve_broadcastrrcCr)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<)rPrjrr<r<r=rls zEagerSeries._with_native2EagerNamespace[Any, Self, Any, Any, NativeSeriesT]cCrMrNr<rOr<r<r=__narwhals_namespace__rz"EagerSeries.__narwhals_namespace__EagerExpr[Any, Any]cCs|j|SrN)r_expr _from_seriesrOr<r<r=rprzEagerSeries._to_exprrows&SizedMultiIndexSelector[NativeSeriesT]cCrMrNr<rPrr<r<r=_gatherrXzEagerSeries._gather_SliceIndex | rangecCrMrNr<rr<r<r= _gather_slicerXzEagerSeries._gather_slicer`racCsJt|ttfr ||St|r||jSt|r||St|dSrN) isinstanceslicerangerrrrUrrrbr<r<r=rcs    zEagerSeries.__getitem__/EagerSeriesStringNamespace[Self, NativeSeriesT]cCrMrNr<rOr<r<r=rSrRzEagerSeries.str1EagerSeriesDateTimeNamespace[Self, NativeSeriesT]cCrMrNr<rOr<r<r=dtrRzEagerSeries.dt,EagerSeriesCatNamespace[Self, NativeSeriesT]cCrMrNr<rOr<r<r=catrRzEagerSeries.cat-EagerSeriesListNamespace[Self, NativeSeriesT]cCrMrNr<rOr<r<r=listrRzEagerSeries.list/EagerSeriesStructNamespace[Self, NativeSeriesT]cCrMrNr<rOr<r<r=struct rRzEagerSeries.structNrLr)rjr#rLr)rrrLr#)rjrrr]rLr#)rLr)rLr)rrrLr#)rrrLr#)r`rarLr#)rLr)rLr)rLr)rLr)rLr)r8r9r:r;rrrrrrlrrprrrcrSrrrrr<r<r<r=rCs6            rCc@sfeZdZUded<edddZedddZedd d Zedd dZedddZ dddZ dS)_SeriesNamespacer_compliant_seriesrLcCs|jSrNrrOr<r<r= compliantsz_SeriesNamespace.compliantr)cC|jjSrN)rrKrOr<r<r=implementationz_SeriesNamespace.implementationrcCrrN)rrrOr<r<r=backend_versionrz _SeriesNamespace.backend_versionr*cCrrN)rr}rOr<r<r=rm rz_SeriesNamespace.versionrcCrrN)rrUrOr<r<r=rU$rz_SeriesNamespace.nativerjrcCs |j|SrN)rrlrkr<r<r= with_native(rz_SeriesNamespace.with_nativeN)rLr)rLr)r)rLr*)rLr)rjrrLr) r8r9r:r;rrrrrmrUrr<r<r<r=r s      rc@s eZdZUded<d ddZdS) rHrrrjrLNonecCs ||_dSrNrrkr<r<r=__init__2rizEagerSeriesNamespace.__init__N)rjrrLr)r8r9r:r;rr<r<r<r=rH,s rHc@ eZdZdS)rDNr8r9r:r<r<r<r=rD6rDc@r)rENrr<r<r<r=rE=rrEc@r)rGNrr<r<r<r=rGDrrGc@r)rINrr<r<r<r=rIKrrIc@r)rJNrr<r<r<r=rJRrrJc@seZdZUded<ded<ded<ed6d d Zed7ddZd8ddZddd9ddZ d:dd Z d;d!d"Z dd,d-Zd?d/d0Zd@d1d2ZdAd3d4Zd5S)BrFEagerSeries[NativeSeriesT]_seriesr] _breakpoint$HistData[NativeSeriesT, _CountsT_co]_datarLrcCrrN)r rUrOr<r<r=rU^rzEagerSeriesHist.nativerjrr#cCs||}||_||_|SrN)__new__r r )rtrjrobjr<r<r= from_seriesbs zEagerSeriesHist.from_seriesrcCrMrNr<rOr<r<r=rkrXzEagerSeriesHist.to_frameboth)closedstartrend num_samplesrfrLiteral['both', 'none']r3cCrMrNr<)rPrrrrr<r<r= _linear_spacelrzEagerSeriesHist._linear_spacecCrMrNr<rOr<r<r=is_empty_seriesvrXzEagerSeriesHist.is_empty_seriescCs|jrggdSdgiS)N)r6r7r7)r rOr<r<r= data_emptyyszEagerSeriesHist.data_emptyargint | list[float]cCrMrNr<)rPrr<r<r= series_empty~rzEagerSeriesHist.series_emptyrrcCsBt|dkr ||_|S|r|||_|S|||_|S)N)rgrr rr_calculate_histrPrr<r<r= with_binss    zEagerSeriesHist.with_binsrcCsD|dkr ||_|S|r|||_|S||||_|Sr)rr rrr_calculate_binsrPrr<r<r=with_bin_counts  zEagerSeriesHist.with_bin_countlist[float] | _1DArraycCs,t|tr|dd|dn|}|ddS)Nrr)rrfr)rPrrr<r<r=_calculate_breakpoints  z%EagerSeriesHist._calculate_breakpointcCrMrNr<r!r<r<r=r rXzEagerSeriesHist._calculate_binscCrMrNr<rr<r<r=rrzEagerSeriesHist._calculate_histNr)rjrrr]rLr#)rLr) rrrrrrfrrrLr3r)rLr )rrrLr )rrrLr#)rrfrLr#)rrrLr#)rrfrLr3)rr#rLr )r8r9r:r;rrUrrrrrrrrr"r$r rr<r<r<r=rFYs&        rF)W __future__rtypingrrrrr!narwhals._compliant.any_namespacerr r r r narwhals._compliant.columnr narwhals._compliant.typingrrrrrnarwhals._translaterrrrnarwhals._typing_compatrrnarwhals._utilsrrrrrcollections.abcrrr typesr!pandaspdpolarsplpyarrowpatyping_extensionsr"r#r$narwhals._compliant.dataframer%narwhals._compliant.exprr&r'narwhals._compliant.namespacer(r)r*r+narwhals.dtypesr,narwhals.seriesr-narwhals.typingr.r/r0r1r2r3r4r5r>__all__rBrCrrHrDrErGrIrJrFr<r<r<r=s       $  x J