O3jddlmZddlZddlmZmZmZddlmZm Z m Z er"ddl Z ddl ZddlmZmZddlmZddlmZddlmZGd d ej0Z dd ZGd d ZddZy)) annotationsN) TYPE_CHECKINGAnyNoReturn)Version_hasattr_static parse_version)SelfTypeIsInterchangeSeries)DType) DataFrameLikec(eZdZdZdZdZdZdZdZdZ y) DtypeKindrN) __name__ __module__ __qualname__INTUINTFLOATBOOLSTRINGDATETIME CATEGORICALL/DATA/.local/lib/python3.12/site-packages/narwhals/_interchange/dataframe.pyrrs% C D E D FHKr#rc8tjj}|dtjk(rm|ddk(r|j S|ddk(r|j S|ddk(r|jS|ddk(r|jSd}t||dtjk(rm|ddk(r|jS|ddk(r|jS|ddk(r|jS|ddk(r|jSd}t||dtjk(r=|ddk(r|j!S|ddk(r|j#Sd }t||dtj$k(r|j'S|dtj(k(r|j+S|dtj,k(r|j/S|dtj0k(r|j3Sd |}t|) Nrr@ zInvalid bit width for INTzInvalid bit width for UINTzInvalid bit width for FLOATzInvalid dtype, got: )rV1dtypesrrInt64Int32Int16Int8AssertionErrorrUInt64UInt32UInt16UInt8rFloat64Float32rBooleanrStringr Datetimer! Categorical)interchange_dtyper+msgs r$'map_interchange_dtype_to_narwhals_dtyper=s ZZ  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!!## !2 3 4C  r#ceZdZejZd dZddZddZddZ ddZ ddZ e ddZ e ddZdd Zdd Zdd Zy )InterchangeFramec.|j|_yN) __dataframe___interchange_frame)selfdfs r$__init__zInterchangeFrame.__init__Ns"$"2"2"4r#c|SrAr"rDs r$__narwhals_dataframe__z'InterchangeFrame.__narwhals_dataframe__Qs r#cd}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)rDr<s r$__native_namespace__z%InterchangeFrame.__native_namespace__Ts Y "#&&r#cPddlm}||jj|S)Nrr )narwhals._interchange.seriesr rCget_column_by_name)rDnamer s r$ get_columnzInterchangeFrame.get_column\s!B !8!8!K!KD!QRRr#cddl}t|dkrd|j}t||jj j |jS)Nr)rrzpConversion to pandas is achieved via interchange protocol which requires 'pandas>=1.5.0' to be installed, found )pandasr __version__rLapi interchangefrom_dataframerC)rDpdr<s r$ to_pandaszInterchangeFrame.to_pandasasZ  y (;;=>>:JL &c* *vv!!001H1HIIr#c2ddlm}||jS)Nr)rY)"pyarrow.interchange.from_dataframerYrC)rDrYs r$to_arrowzInterchangeFrame.to_arrowls d5566r#c |jjDcic]1}|t|jj|j3c}Scc}wrA)rC column_namesr=rPdtype)rD column_names r$schemazInterchangeFrame.schemasse $66CCE   F  @''::;GMM  F    s6AcHt|jjSrA)listrCr`rHs r$columnszInterchangeFrame.columns|sD++88:;;r#c$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.rK)rDattrr<s r$ __getattr__zInterchangeFrame.__getattr__s'B B "#&&r#c|jjt|}t|ds d}t ||j |j S)N_dfzExpected interchange object to implement `_df` property to allow for recovering original object. See https://github.com/data-apis/dataframe-api/issues/360.)rCselect_columns_by_namerehasattrrL __class__rk)rDr`framer<s r$ simple_selectzInterchangeFrame.simple_selectsR''>>tL?QRue$M &c* *~~eii((r#cd}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.rK)rDexprsr<s r$selectzInterchangeFrame.selects B "#&&r#N)rErreturnNone)rtr )rtr)rQstrrtr )rtz pd.DataFrame)rtzpa.Table)rtzdict[str, DType])rtz list[str])rhrvrtr)r`rvrtr )rrrvrtr )rrrrr*_versionrFrIrMrRr[r^propertyrcrfrirprsr"r#r$r?r?KsbzzH5'S J7  <<')'r#r?ct|dS)NrB)r)objs r$supports_dataframe_interchanger{s 3 00r#)r;ztuple[DtypeKind, int, Any, Any]rtr)rzrrtzTypeIs[DataFrameLike]) __future__renumtypingrrrnarwhals._utilsrrr rUrZpyarrowpatyping_extensionsr r rOr narwhals.dtypesrnarwhals.stable.v1.typingrIntEnumrr=r?r{r"r#r$rsd" //CC.>%7 +6+ +\P'P'f1r#