hhddlmZddlZddlmZmZmZddlmZm Z er"ddl Z ddl Z ddlmZmZddlmZddlmZddlmZGd d ejZddZGddZddZdS)) annotationsN) TYPE_CHECKINGAnyNoReturn)Version parse_version)SelfTypeIsInterchangeSeries)DType) DataFrameLikec*eZdZdZdZdZdZdZdZdZ dS) DtypeKindrN) __name__ __module__ __qualname__INTUINTFLOATBOOLSTRINGDATETIME CATEGORICALV/var/www/histauto/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)(InterchangeFramedfrr&Nonec8||_dSN) __dataframe___interchange_frame)selfrAs r#__init__zInterchangeFrame.__init__Ns"$"2"2"4"4r"r c|SrDr!rGs 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)rGr=s r#__native_namespace__z%InterchangeFrame.__native_namespace__Ts Y "#&&&r"namestrr cTddlm}||j|S)Nrr )narwhals._interchange.seriesr rFget_column_by_name)rGrPr s r# get_columnzInterchangeFrame.get_column\s6BBBBBB  !8!K!KD!Q!QRRRr" pd.DataFramecddl}t|dkrd|j}t||jj|jS)Nr)rrzpConversion to pandas is achieved via interchange protocol which requires 'pandas>=1.5.0' to be installed, found )pandasr __version__rNapi interchangefrom_dataframerF)rGpdr=s r# to_pandaszInterchangeFrame.to_pandasasi   y ( (L;=>LL &c** *v!001HIIIr"pa.Tablec.ddlm}||jS)Nr)r])"pyarrow.interchange.from_dataframer]rF)rGr]s r#to_arrowzInterchangeFrame.to_arrowls5      ~d5666r"dict[str, DType]cNfdjDS)Ncji|]/}|tj|j0Sr!)r>rFrTdtype).0 column_namerGs r# z+InterchangeFrame.schema..usO    @'::;GGM   r")rF column_namesrJs`r#schemazInterchangeFrame.schemass?     $6CCEE     r" list[str]cNt|jSrD)listrFrkrJs 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.rM)rGrqr=s r# __getattr__zInterchangeFrame.__getattr__s1 B B B B "#&&&r"rkc|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.)rFselect_columns_by_namerohasattrrN __class__ru)rGrkframer=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.rM)rGr{r=s r#selectzInterchangeFrame.selects B "#&&&r"N)rArr&rB)r&r )r&r)rPrQr&r )r&rV)r&r`)r&rd)r&rm)rqrQr&r)rkrQr&r )r{rQr&r )rrrrr,_versionrHrKrOrUr_rcpropertyrlrprsrzr}r!r"r#r@r@KszH5555''''SSSS J J J J7777   X <<<X<''''))))''''''r"r@objrTypeIs[DataFrameLike]c"t|dS)NrE)rw)rs r#supports_dataframe_interchangers 3 ( ((r")r$r%r&r )rrr&r) __future__renumtypingrrrnarwhals._utilsrrrYr^pyarrowpatyping_extensionsr r rSr narwhals.dtypesr narwhals.stable.v1.typingrIntEnumrr>r@rr!r"r#rsT"""""" //////////222222228........>>>>>>%%%%%%777777 ++++\P'P'P'P'P'P'P'P'f))))))r"