7^jddlmZddlZddlmZmZmZddlmZm 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 CATEGORICALj/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/narwhals/_interchange/dataframe.pyrrs% C D E D FHKr#rchtjj}|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(rU|ddk(r|j!S|ddk(r|j#S|ddk(r|j%Sd }t||dtj&k(r|j)S|dtj*k(r|j-S|dtj.k(r|j1S|dtj2k(r|j5Sd |}t|) Nrr@ zInvalid bit width for INTzInvalid bit width for UINTzInvalid bit width for FLOATzInvalid dtype, got: )rV1dtypesrrInt64Int32Int16Int8AssertionErrorrUInt64UInt32UInt16UInt8rFloat64Float32Float16rBooleanrStringr 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 %>># # 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__Ps"$"2"2"4r#c|SrBr"rEs r$__narwhals_dataframe__z'InterchangeFrame.__narwhals_dataframe__Ss 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)rEr=s r$__native_namespace__z%InterchangeFrame.__native_namespace__Vs Y "#&&r#cPddlm}||jj|S)Nrr )narwhals._interchange.seriesr rDget_column_by_name)rEnamer 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__rMapi interchangefrom_dataframerD)rEpdr=s r$ to_pandaszInterchangeFrame.to_pandascsZ  y (;;=>>:JL &c* *vv!!001H1HIIr#c2ddlm}||jS)Nr)rZ)"pyarrow.interchange.from_dataframerZrD)rErZs r$to_arrowzInterchangeFrame.to_arrowns d5566r#c |jjDcic]1}|t|jj|j3c}Scc}wrB)rD column_namesr>rQdtype)rE column_names r$schemazInterchangeFrame.schemaus\ $66CCE   @''::;GMM    s6AcHt|jjSrB)listrDrarIs 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.rL)rEattrr=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.)rDselect_columns_by_namerfhasattrrM __class__rl)rEraframer=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.rL)rEexprsr=s r$selectzInterchangeFrame.selects B "#&&r#N)rFrreturnNone)rur )rur)rRstrrur )ruz pd.DataFrame)ruzpa.Table)ruzdict[str, DType])ruz list[str])rirwrur)rarwrur )rsrwrur )rrrrr*_versionrGrJrNrSr\r_propertyrdrgrjrqrtr"r#r$r@r@MsbzzH5'S J7  <<')'r#r@ct|dS)NrC)r)objs r$supports_dataframe_interchanger|s 3 00r#)r<ztuple[DtypeKind, int, Any, Any]rur)r{rruzTypeIs[DataFrameLike]) __future__renumtypingrrrnarwhals._utilsrrr rVr[pyarrowpatyping_extensionsr r rPr narwhals.dtypesrnarwhals.stable.v1.typingrIntEnumrr>r@r|r"r#r$rsd" //CC.>%7 -6- -`P'P'f1r#