}i UddlmZddlmZddlmZmZmZmZm Z m Z m Z ddl Z ddl mZmZddlmZddlmZmZmZddlmZmZdd lmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&dd l'm(Z)m*Z+dd l,m-Z-m.Z.dd l/m0Z1dd l2m3Z3m4Z4m5Z5ddl6m7Z8ddl9m:Z;ddlZ>m?Z?ddl@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\ddl]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZeddlfmgZgmhZhmiZierddljmkZkmlZlmmZmddlnmoZoddlpmqZqmrZrmsZsddlmtZtddlumvZwmxZymzZ{m|Z}m~ZmZmZmZmZmZmZmZmZmZmZddlmZmZmZmZmZmZmZddl'mZmZddlmZddlmZmZmZmZmZmZmZmZmZmZedeZeqdZed ZGd!d"e)e_Z(Gd#d$e+eaZ*Gd%d&e;ecZ:Gd'd(e1Z0Gd)d*e8Z7e dd/Ze dd2Ze dd5Ze dd7Zdd:Ze dd?Ze ddAZe ddCZe ddEZe ddHZe ddJZe ddMZe ddOZe ddRZe ddTZe ddUZe ddVZe ddXZe ddYZe dd\Ze dd]Ze dd_Ze dd`Ze dddZe ddfZe ddhZe ddjZe dkdldduZdddkdkdkddvddzZe d{d|ddZe d{d|ddZe d{d|ddZe ddZe d{dd dZe d{dd dZe d{dd dZe d dZdddd dZ ddddkdkdkddvddZddZddZddZddZddZdddZddZddZddZddZddZddZdkdddZdkdddZddZddZddZddkdddZddZddZddZGddejZGddeje0ZŐdd„Zedæ ddddĜdd̄ZedædddĜddЄZe ddddĜddՄZe(jZded<edæ ddddĜddڄZedædddĜd dބZedædddĜd!dZedædddĜd dZedædddĜd!dZАd"dZgdZdS(#) annotationswraps) TYPE_CHECKINGAnyCallableFinalLiteralcastoverloadN) exceptions functions) issue_warning)ExprKindExprNodeis_expr)TypeVar assert_never) ImplementationVersiondeprecate_native_namespacegenerate_temporary_column_name inherit_docis_ordered_categoricalmaybe_align_indexmaybe_convert_dtypesmaybe_get_indexmaybe_reset_indexmaybe_set_indexvalidate_strict_and_pass_though) DataFrame LazyFrame)InvalidIntoExprErrorNarwhalsUnstableWarning)Expr)_new_series_implconcat show_versions)Schema)Series) dependenciesdtypes selectors)ArrayBinaryBoolean CategoricalDateDatetimeDecimalDurationEnumFieldFloat32Float64Int8Int16Int32Int64Int128ListObjectStringStructTimeUInt8UInt16UInt32UInt64UInt128Unknown) DataFrameTIntoDataFrameT IntoFrameIntoLazyFrameT IntoSeries IntoSeriesT LazyFrameTSeriesT)_from_native_implget_native_namespace to_py_scalar)IterableMappingSequence) ModuleType) ParamSpecSelfUnpack) ExprMetadata)AllowAnyStrictV1 AllowAnyV1AllowLazyStrictV1 AllowLazyV1AllowSeriesStrictV1 AllowSeriesV1ExcludeSeriesStrictV1ExcludeSeriesV1IntoArrowTableOnlyEagerOrInterchangeOnlyEagerOrInterchangeStrictOnlySeriesStrictV1 OnlySeriesV1PassThroughUnknownV1StrictUnknownV1)ArrowBackend EagerAllowed IntoBackend LazyAllowedPandasPolars)MultiColSelectorMultiIndexSelector)DType) FileSource IntoDTypeIntoExpr IntoSchemaNonNestedLiteral PythonLiteralSingleColSelectorSingleIndexSelector_1DArray_2DArrayT)defaultPRceZdZejZeedFfd Ze dGfd Z e dHdddIfdZ e dHdJfd Z e dHdKfd Z edLdZedMdZedNd!ZedOd$ZedPd'ZdQfd* ZdRfd. Z dHdd/dSfd4Zed5d6dTd:ZedUd=Zed>d6dVdAZd>d6dVfdBZdWfdC ZdWfdD ZdXdEZxZS)Yr!dfrlevel&Literal['full', 'lazy', 'interchange']returnNonecx|jtjusJt||dSN)r_versionrV1super__init__selfrr __class__s O/home/jrussi/.local/lib/python3.11/site-packages/narwhals/stable/v1/__init__.pyrzDataFrame.__init__;{gj(((( 5))))) native_framerebackendIntoBackend[EagerAllowed]DataFrame[Any]cjt||}td|SNrr)r from_arrowr )clsrrresultrs rrzDataFrame.from_arrows2##L'#BB$f---rNrdataMapping[str, Any]schema.IntoSchema | Mapping[str, DType | None] | None IntoBackend[EagerAllowed] | Noneclt|||}td|Sr)r from_dictr rrrrrrs rrzDataFrame.from_dicts4""4"AA$f---r Sequence[Any]clt|||}td|Sr)r from_dictsr rs rrzDataFrame.from_dicts4##D&'#BB$f---rr3Mapping[str, DType] | Schema | Sequence[str] | Noneclt|||}td|Srr from_numpyr rs rrzDataFrame.from_numpyrrtype[Series[Any]]c,tdtS)Nr)r r*rs r_serieszDataFrame._seriess'000rtype[LazyFrame[Any]]c,tdtS)Nr)r r"rs r _lazyframezDataFrame._lazyframes*I666ritem-tuple[SingleIndexSelector, SingleColSelector]cdSNrrs r __getitem__zDataFrame.__getitem__WZWZr2str | tuple[MultiIndexSelector, SingleColSelector] Series[Any]cdSrrrs rrzDataFrame.__getitem__s crSingleIndexSelector | MultiIndexSelector | MultiColSelector | tuple[SingleIndexSelector, MultiColSelector] | tuple[MultiIndexSelector, MultiColSelector]rZcdSrrrs rrzDataFrame.__getitem__s sr SingleIndexSelector | SingleColSelector | MultiColSelector | MultiIndexSelector | tuple[SingleIndexSelector, SingleColSelector] | tuple[SingleIndexSelector, MultiColSelector] | tuple[MultiIndexSelector, SingleColSelector] | tuple[MultiIndexSelector, MultiColSelector]Series[Any] | Self | AnycFt|Sr)rr)rrrs rrzDataFrame.__getitem__sww""4(((rnamestrr*cFt|Sr)r get_column)rrrs rrzDataFrame.get_columnsww!!$'''r)sessionIntoBackend[LazyAllowed] | Noner Any | NoneLazyFrame[Any]cdtt||S)N)rr) _stableifyrlazy)rrrrs rrzDataFrame.lazys' %'',,w,HHIIIr. as_seriesr Literal[True]dict[str, Series[Any]]cdSrrrrs rto_dictzDataFrame.to_dictsTWTWrLiteral[False]dict[str, list[Any]]cdSrrrs rrzDataFrame.to_dictsMPSrTbool-dict[str, Series[Any]] | dict[str, list[Any]]cdSrrrs rrzDataFrame.to_dicts 9<rcHt|S)Nr)rr)rrrs rrzDataFrame.to_dicts ww333rc^ttSr)rr is_duplicatedrrs rrzDataFrame.is_duplicateds!%''//11222rc^ttSr)rr is_uniquers rrzDataFrame.is_uniques!%''++--...rch|tSrselectall_l1_normrs rrzDataFrame._l1_norm"{{355>>++,,,rrrrrrr)rrerrrrr)rrrrrrrr)rrrrrrrr)rrrrrrrr)rr)rr)rrrr)rrrr)rrrrZ)rrrr)rrrr*)rrrrrr)rrrr)rrrr)rrrr)rrrrZ)__name__ __module__ __qualname__rrrr NwDataFramer classmethodrrrrpropertyrrr rrrrrrr __classcell__rs@rr!r!s%zH[******.....[. BF. 59 .......[.BF......[.GK......[.111X1777X7ZZZXZ X   X  ) ) ) ) ) )((((((48J# JJJJJJJJ47WWWWWXW PPPXP #'<<<<<X<$(44444444333333//////--------rr!ceZdZejZeed#fd Ze d$d Z d%d Z d&d'fd Z d(dZ d)d*fd Zd+d,dZ d-d dd.d"ZxZS)/r"rrrrrrcx|jtjusJt||dSrrrs rrzLazyFrame.__init__rrtype[DataFrame[Any]]ctSrr!rs r _dataframezLazyFrame._dataframe rmetadatar\cdSrr)rrs r_validate_metadatazLazyFrame._validate_metadatas rNr+IntoBackend[Polars | Pandas | Arrow] | Nonekwargsrc Tttjdd|i|S)Nrr)rrcollect)rrrrs rrzLazyFrame.collects./%''/DD'DVDDEEErrZch|tSrrrs rrzLazyFrame._l1_normrrnintcFt|S)Get the last `n` rows.)rtail)rrrs rr zLazyFrame.tailsww||Arroffsetc`||j||S)zTake every nth row in the DataFrame and return as a new DataFrame. Arguments: n: Gather every *n*-th row. offset: Starting index. rr )_with_compliant_compliant_frame gather_everyrrr s rrzLazyFrame.gather_every!s5##  ! . .6 . B B   rindex)order_byrrrstr | Sequence[str] | Nonect|tr|gn|}||j||S)N)rr) isinstancerrrwith_row_index)rrr order_by_s rrzLazyFrame.with_row_index,sW#-Xs";";IXJJ ##  ! 0 0" 1     rrrr)rr\rrr)rrrrrrr)rrrrrZrrrr rrrZ)r)rrrrrrZ)rrrrrrr NwLazyFramerrrrrrr rrrrs@rr"r"s9zH[******X    FJFFFFFFF ----      "  MQ           rr"ceZdZejZeed-fd Ze d.d/fd Z e d.d0fd Z e d1dZ d2fd Zdddddd3fd Z d.dd!d"d4fd(Zdd)d5fd,ZxZS)6r*seriesrrrrrcx|jtjusJt||dSrr)rrrrs rrzSeries.__init__;s='*,,,, u-----rNrrvaluesr~dtypeIntoDType | Nonerrrcnt||||}td|SNrrrrrr!r"rrrs rrzSeries.from_numpyEs5##D&%#IIM6***r Iterable[Any]cnt||||}td|Sr%)r from_iterabler r&s rr)zSeries.from_iterableQs5&&tVUG&LLM6***rrctSrrrs rrzSeries._dataframe]rrrc^ttSr)rrto_framers rr,zSeries.to_frameas!%''**,,---rFsortparallelr normalizer.rr/ str | Noner0chtt||||S)Nr-)rr value_counts)rr.r/rr0rs rr3zSeries.value_countsds< GG H49 !     rT) bin_countinclude_breakpointbinslist[float] | Noner4 int | Noner5cd}t|ttt|||S)NzZ`Series.hist` is being called from the stable API although considered an unstable feature.)r6r4r5)rr$rrhist)rr6r4r5msgrs rr:z Series.histrsT #  c2333 GGLLYCU      r ignore_nullsr=r{cvd}t|tt|S)Nz_`Series.any_value` is being called from the stable API although considered an unstable feature.r<)rr$r any_value)rr=r;rs rr?zSeries.any_values; #  c2333ww  l ;;;r)rrrrrrr) rrr!r~r"r#rrrr) rrr!r'r"r#rrrrr)rr) r.rr/rrr1r0rrr)r6r7r4r8r5rrr)r=rrr{)rrrrrrrNwSeriesrrrr)rrr,r3r:r?rrs@rr*r*8szH[...... #' + + + + + +[ + #' + + + + + +[ +X......          $( !%#'         $16<<<<<<<<<<<zBnarwhalify..decorator..wrapper..sL   !-).G +!-    rc Bi|]\}}|t|Srr)rrvaluerrrrrs r zBnarwhalify..decorator..wrapper..sR    D%k!-).G +!-    rcJh|]}t|ddx S)__native_namespace__N)getattr)rvbs r zAnarwhalify..decorator..wrapper..sG $:DAAAArz_Found multiple backends. Make sure that all dataframe/series inputs come from the same backend.r)itemsr!__len__ ValueErrorr)rrargs_nw kwargs_nwbackendsr;rrrrrrrrs @rwrapperz.narwhalify..decorator..wrappers            G        $*<<>>   I878Y%5%5%7%788H !!A%%w oo%T70i00FV,??? ?r)rrrrrrr)rrrrrrrs` r decoratorznarwhalify..decorators^ t% @% @% @% @% @% @% @% @% @% @ % @NrN)rrrrr)rrrrrrrrs ````` r narwhalifyrsr 3 4L))))))))))V | 9T??rcBttjSr)rrrrrrrr bfhh  rnamesstr | Iterable[str]c8ttj|Sr)rrcolrs rrrs bfen % %%rc8ttj|Sr)rrexcluders rrrs bj%( ) ))rindicesint | Sequence[int]c8ttj|Sr)rrnth)rs rr r # bfg& ' ''rcBttjSr)rrlenrrrr r 'rrrrzr"r#cFttj||Sr)rrlit)rr"s rrr+s bfUE** + ++rcolumnsrc8ttj|Sr)rrminrs rrr/r rc8ttj|Sr)rrmaxrs rrr3r rc8ttj|Sr)rrmeanrs rrr7s bgw' ( ((rc8ttj|Sr)rrmedianrs rrr;s bi) * **rc8ttj|Sr)rrsumrs rrr?r rexprsIntoExpr | Iterable[IntoExpr]c8ttj|Sr)rrsum_horizontalrs rrrC b'/ 0 00rr<r=c>ttj|d|iSNr=)rrall_horizontalr=rs rr#r#G" b'J\JJ K KKrc>ttj|d|iSr")rrany_horizontalr$s rr'r'Mr%rc8ttj|Sr)rrmean_horizontalrs rr)r)Ss b(%0 1 11rc8ttj|Sr)rrmin_horizontalrs rr+r+Wr rc8ttj|Sr)rrmax_horizontalrs rr-r-[r r separatorr= more_exprsrxr0cHttj|g|R||dS)Nr/)rr concat_str)rr0r=r1s rr3r3_s5  eYjYYILYYY  rf_stringrc@ttj|g|RS)zFormat expressions as a string.)rrformat)r4rs rr6r6js" bi04000 1 11rc@ttj|g|RSr)rrcoalesce)rr1s rr8r8os" bk%5*555 6 66r5DataFrame[Any] | LazyFrame[Any] | Series[IntoSeriesT]rc|jS)aLevel of support Narwhals has for current object. Arguments: obj: Dataframe or Series. Returns: This can be one of - 'full': full Narwhals API support - 'lazy': only lazy operations are supported. This excludes anything which involves iterating over rows in Python. - 'interchange': only metadata operations are supported (`df.schema`) )rrws r get_levelr;ss :rc.eZdZed dZd dZd S) Whenwhen nw_f.Whenrc&||jdS)Nr)chain) _predicate)rr>s r from_whenzWhen.from_whenss4?"----rr&IntoExpr | NonNestedLiteral | _1DArrayThenc\g|j|j|fR}t|Sr)_chainrBrE _from_chain)rr new_chains rthenz When.thens1r?rr=)rrDrrE)rrrrrCrJrrrr=r=sF...[.++++++rr=ceZdZdS)rEN)rrrrrrrErEsrrE predicatescNttj|Sr)r=rCnw_fr>)rLs rr>r>s >>$)Z0 1 11r)required)rnative_namespacerr!rrrPModuleType | Nonercbtd|}tt||||S)a=Instantiate Narwhals Series from iterable (e.g. list or array). See `narwhals.new_series` for full docstring. Note that `native_namespace` is an is the same as `backend` but only accepts module types - for new code, we recommend using `backend`, as that's available beyond just `narwhals.stable.v1`. rr)r rr&)rr!r"rrPs r new_seriesrSs4.88G &tVUGLLL M MMrrrerchtd|}ttj||S)aJConstruct a DataFrame from an object which supports the PyCapsule Interface. See `narwhals.from_arrow` for full docstring. Note that `native_namespace` is an is the same as `backend` but only accepts module types - for new code, we recommend using `backend`, as that's available beyond just `narwhals.stable.v1`. rr)r rrNr)rrrPs rrrs1.88G dolGDDD E EErrrr#Mapping[str, DType] | Schema | NonecJttj|||S)a#Instantiate DataFrame from dictionary. See `narwhals.from_dict` for full docstring. Note that `native_namespace` is an is the same as `backend` but only accepts module types - for new code, we recommend using `backend`, as that's available beyond just `narwhals.stable.v1`. r)rrNrrrrrPs rrrs# dnT67CCC D DDrr rrrcjtd|}ttj|||S)a)Construct a DataFrame from a NumPy ndarray. See `narwhals.from_numpy` for full docstring. Note that `native_namespace` is an is the same as `backend` but only accepts module types - for new code, we recommend using `backend`, as that's available beyond just `narwhals.stable.v1`. rr)r rrNrrWs rrrs3.88G dodFGDDD E EErsourcervrc dtd|}ttj|fd|i|S)aRead a CSV file into a DataFrame. See `narwhals.read_csv` for full docstring. Note that `native_namespace` is an is the same as `backend` but only accepts module types - for new code, we recommend using `backend`, as that's available beyond just `narwhals.stable.v1`. rr)r rrNread_csvrYrrPrs rr[r[s9.88G dmFFFGFvFF G GGrIntoBackend[Backend] | Nonerc dtd|}ttj|fd|i|S)aLazily read from a CSV file. See `narwhals.scan_csv` for full docstring. Note that `native_namespace` is an is the same as `backend` but only accepts module types - for new code, we recommend using `backend`, as that's available beyond just `narwhals.stable.v1`. IntoBackend[Backend]r)r rrNscan_csvr\s rr`r`s9)733G dmFFFGFvFF G GGrc dtd|}ttj|fd|i|S)a*Read into a DataFrame from a parquet file. See `narwhals.read_parquet` for full docstring. Note that `native_namespace` is an is the same as `backend` but only accepts module types - for new code, we recommend using `backend`, as that's available beyond just `narwhals.stable.v1`. rr)r rrN read_parquetr\s rrbrbs:.88G d'JJJ6JJ K KKrc dtd|}ttj|fd|i|S)a Lazily read from a parquet file. See `narwhals.scan_parquet` for full docstring. Note that `native_namespace` is an is the same as `backend` but only accepts module types - for new code, we recommend using `backend`, as that's available beyond just `narwhals.stable.v1`. r_r)r rrN scan_parquetr\s rrdrds:)733G d'JJJ6JJ K KKrIntoExpr | Sequence[IntoExpr] named_exprsc>ttj|i|S)aCollect columns into a struct column. Arguments: *exprs: Column(s) to collect into a struct column, specified as positional arguments. Accepts only expression input. Strings are parsed as column names, other non-expression inputs are not allowed. **named_exprs: Additional columns to collect into the struct column, specified as keyword arguments. The columns will be renamed to the keyword used. )rrNstruct)rrfs rrhrh)s! dk58K88 9 99r)Vr.r/r0r1r!r2r3r4r5r6r%r7r8r9rr:r;r<r=r>r#r"r?r@r)r*rArBrCrDrErFrGrHrIrr#r'r8rr'r3r+r,r rr6rrrrrrr;rSrrr rrr-rrrrrrr)rrr+rrSr r[rbr`rdr-r(rhrrrrTr>)rsrtrru)rsryrrz)rsr|rr})rsrrr%)rsrrr)rrQrrrrQ)rrQrrrrQ)rrQrrrrQ)rrQrrrrQ)rrJrrrrJ)rrJrrrrJ)rrPrrrrP)rrPrrrrP)rrKrrrru)rrKrrrru)rrKrrrru)rrKrrrru)rrOrrrr})rrOrrrr})rrrrrr)rrrrrr)rrMrrrrz)rrMrrrrz)rrrrrr)rrrrrr)rrrrrr)rrrrrr)rrrrrrrrrrrrPrr)rrrrPrrPrrrrrrrrPrrrr)rrurrrrK)rrzrrrrM)rr}rrrrO)rrrrrr)rrurrrrK)rrzrrrrM)rr}rrrrO)rrrrrr)rrrrPrrPrrr)rrrrPrrPrrrrrrrrPrr)rr%)rrrr%)rrrr%)rrzr"r#rr%)rrrr%)rrrr%)rrr=rrr%) rrr1rxr0rr=rrr%)r4rrrxrr%)rrr1rxrr%)rsr9rr)rLrrr=) rrr!rr"r#rrrPrQrr)rrerrrPrQrr) rrrrUrrrPrQrr) rrrrrrrPrQrr) rYrvrrrPrQrrrr) rYrvrr]rPrQrrrr)rrerfrxrr%) __future__r functoolsrtypingrrrr r r r narwhalsrr rrNnarwhals._exceptionsrnarwhals._expression_parsingrrrnarwhals._typing_compatrrrrrrrrrrrrrrr narwhals.dataframer!rr"rnarwhals.exceptionsr#r$ narwhals.exprr%rnarwhals.functionsr&r'r(narwhals.schemar)rrnarwhals.seriesr*r@narwhals.stable.v1r+r,r-narwhals.stable.v1.dtypesr.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErFrGrHrInarwhals.stable.v1.typingrJrKrLrMrNrOrPrQnarwhals.translaterRrSrTcollections.abcrUrVrWtypesrXtyping_extensionsrYrZr[r\narwhals._translater]AllowAnyStrictr^AllowAnyr_AllowLazyStrictr` AllowLazyraAllowSeriesStrictrb AllowSeriesrcExcludeSeriesStrictrd ExcludeSeriesrerfrgrhOnlySeriesStrictri OnlySeriesrjPassThroughUnknownrk StrictUnknownnarwhals._typingrlrmrnrorprqrrrsrtnarwhals.dtypesrunarwhals.typingrvrwrxryrzr{r|r}r~rrrrrrrrrrrr r rrrrrrrr#r'r)r+r-r3r6r8r;r=rEr>rSrrr__annotations__rr[r`rbrdrh__all__rrrrs"""""""OOOOOOOOOOOOOOOOOO22222222......DDDDDDDDDD99999999                            RQQQQQQQMMMMMMMM((((((FFFFFFFFFF............>>>>>>>>>><                    UTTTTTTTTT2;;;;;;;;;;      9999999999999999"HGGGGGGG%%%%%%                         S!!!A #A AA-A-A-A-A- N+A-A-A-H1 1 1 1 1 N+1 1 1 hR<R<R<R<R