-Ph!ddlmZddlZddlmZmZmZddlmZm Z er ddl Z ddl Z ddlmZddlmZddlmZddlmZGd d ejZddZGddZdS)) annotationsN) TYPE_CHECKINGAnyNoReturn)Version parse_version)SelfInterchangeSeries)DType) DataFrameLikec*eZdZdZdZdZdZdZdZdZ dS) DtypeKindrN) __name__ __module__ __qualname__INTUINTFLOATBOOLSTRINGDATETIME CATEGORICAL_/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/narwhals/_interchange/dataframe.pyrrs1 C D E D FHKKKr!rinterchange_dtypetuple[DtypeKind, int, Any, Any]returnr ctjj}|dtjkr|ddkr|S|ddkr|S|ddkr|S|ddkr|Sd}t||dtj kr|ddkr| S|ddkr| S|ddkr| S|ddkr|Sd}t||dtjkrQ|ddkr|S|ddkr|Sd }t||dtjkr|S|dtjkr|S|dtjkr|S|dtjkr|Sd |}t|) Nrr@ zInvalid bit width for INTzInvalid bit width for UINTzInvalid bit width for FLOATzInvalid dtype, got: )rV1dtypesrrInt64Int32Int16Int8AssertionErrorrUInt64UInt32UInt16UInt8rFloat64Float32rBooleanrStringrDatetimer Categorical)r#r,msgs r"'map_interchange_dtype_to_narwhals_dtyper=sTZ Fy},, Q 2 % %<<>> ! Q 2 % %<<>> ! Q 2 % %<<>> ! Q 1 $ $;;== )S!!!y~-- Q 2 % %==?? " Q 2 % %==?? " Q 2 % %==?? " Q 1 $ $<<>> !*S!!!y.. Q 2 % %>>## # Q 2 % %>>## #+S!!!y~--~~y///}}y111   y444!!### 4!2 4 4C   r!ceZdZejZddZddZdd Zd d Z d!dZ d"dZ e d#dZ e d$dZd%dZd&dZd'dZdS)(InterchangeFramedfr r%Nonec8||_dSN) __dataframe___interchange_frame)selfr@s r"__init__zInterchangeFrame.__init__Ns"$"2"2"4"4r!r c|SrCr rFs r"__narwhals_dataframe__z'InterchangeFrame.__narwhals_dataframe__Qs r!rc$d}t|)NzCannot access native namespace for interchange-level dataframes with unknown backend.If you would like to see this kind of object supported in Narwhals, please open a feature request at https://github.com/narwhals-dev/narwhals/issues.NotImplementedError)rFr<s r"__native_namespace__z%InterchangeFrame.__native_namespace__Ts Y "#&&&r!namestrr cTddlm}||j|S)Nrr )narwhals._interchange.seriesr rEget_column_by_name)rFrOr s r" get_columnzInterchangeFrame.get_column\s6BBBBBB  !8!K!KD!Q!QRRRr! pd.DataFramecddl}t|dkr$|jj|jSd|j}t|)Nr)rrzpConversion to pandas is achieved via interchange protocol which requires 'pandas>=1.5.0' to be installed, found )pandasrapi interchangefrom_dataframerE __version__rM)rFpdr<s r" to_pandaszInterchangeFrame.to_pandasasi    ) )6%44T5LMM ML;=>LL &c** *r!pa.Tablec.ddlm}||jS)Nr)r[)"pyarrow.interchange.from_dataframer[rE)rFr[s r"to_arrowzInterchangeFrame.to_arrowms5      ~d5666r!dict[str, DType]cNfdjDS)Ncji|]/}|tj|j0Sr )r=rErSdtype).0 column_namerFs r" z+InterchangeFrame.schema..vsO    @'::;GGM   r!)rE column_namesrIs`r"schemazInterchangeFrame.schemats?     $6CCEE     r! list[str]cNt|jSrC)listrErjrIs r"columnszInterchangeFrame.columns}sD+88::;;;r!attrc,d|d}t|)Nz Attribute aK is not supported for interchange-level dataframes. Hint: you probably called `nw.from_native` on an object which isn't fully supported by Narwhals, yet implements `__dataframe__`. If you would like to see this kind of object supported in Narwhals, please open a feature request at https://github.com/narwhals-dev/narwhals/issues.rL)rFrpr<s r" __getattr__zInterchangeFrame.__getattr__s1 B B B B "#&&&r!rjc|jt|}t|dsd}t |||jS)N_dfzExpected interchange object to implement `_df` property to allow for recovering original object. See https://github.com/data-apis/dataframe-api/issues/360.)rEselect_columns_by_namernhasattrrM __class__rt)rFrjframer<s r" simple_selectzInterchangeFrame.simple_selectsa'>>tL?Q?QRRue$$ +M &c** *~~ei(((r!exprsc$d}t|)Nz`select`-ing not by name is not supported for interchange-only level. If you would like to see this kind of object better supported in Narwhals, please open a feature request at https://github.com/narwhals-dev/narwhals/issues.rL)rFrzr<s r"selectzInterchangeFrame.selects B "#&&&r!N)r@r r%rA)r%r )r%r)rOrPr%r )r%rU)r%r_)r%rc)r%rl)rprPr%r)rjrPr%r )rzrPr%r )rrrrr+_versionrGrJrNrTr^rbpropertyrkrorrryr|r r!r"r?r?KszH5555''''SSSS + + + +7777   X <<<X<''''))))''''''r!r?)r#r$r%r ) __future__renumtypingrrrnarwhals._utilsrrrXr]pyarrowpatyping_extensionsr rRr narwhals.dtypesr narwhals.typingr IntEnumrr=r?r r!r"rs;"""""" //////////22222222.&&&&&&>>>>>>%%%%%%------ ++++\Q'Q'Q'Q'Q'Q'Q'Q'Q'Q'r!