o Lh@sddlmZddlZddlmZmZmZddlmZm Z er@ddl Z ddl Z ddlmZmZddlmZddlmZddlmZGd d d ejZdddZGdddZdddZdS)) annotationsN) TYPE_CHECKINGAnyNoReturn)Version parse_version)SelfTypeIsInterchangeSeries)DType) DataFrameLikec@s(eZdZdZdZdZdZdZdZdZ dS) DtypeKindrN) __name__ __module__ __qualname__INTUINTFLOATBOOLSTRINGDATETIME CATEGORICALrr`/var/www/html/Persson_Maskin/env/lib/python3.10/site-packages/narwhals/_interchange/dataframe.pyrsrinterchange_dtypetuple[DtypeKind, int, Any, Any]returnr cCstjj}|dtjkr9|ddkr|S|ddkr|S|ddkr)|S|ddkr3|Sd}t ||dtj krn|ddkrJ| S|ddkrT| S|ddkr^| S|ddkrh|Sd}t ||dtjkr|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)msgrrr 'map_interchange_dtype_to_narwhals_dtypesP           r:c@seZdZejZd(ddZd)dd Zd*d d Zd+ddZ d,ddZ d-ddZ e d.ddZ e d/ddZd0dd Zd1d"d#Zd2d%d&Zd'S)3InterchangeFramedfr r#NonecCs||_dSN) __dataframe___interchange_frame)selfr<rrr __init__NszInterchangeFrame.__init__rcCs|Sr>rrArrr __narwhals_dataframe__Qsz'InterchangeFrame.__narwhals_dataframe__rcCs 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)rAr9rrr __native_namespace__Tsz%InterchangeFrame.__native_namespace__namestrr cCsddlm}||j|S)Nrr )narwhals._interchange.seriesr r@get_column_by_name)rArHr rrr get_column\s zInterchangeFrame.get_column pd.DataFramecCs8ddl}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__rFapi interchangefrom_dataframer@)rApdr9rrr to_pandasas zInterchangeFrame.to_pandaspa.TablecCsddlm}||jS)Nr)rS)"pyarrow.interchange.from_dataframerSr@)rArSrrr to_arrowls  zInterchangeFrame.to_arrowdict[str, DType]csfddjDS)Ncs i|] }|tj|jqSr)r:r@rKdtype).0 column_namerCrr us  z+InterchangeFrame.schema..)r@ column_namesrCrrCr schemass zInterchangeFrame.schema list[str]cCst|jSr>)listr@r^rCrrr columns|szInterchangeFrame.columnsattrcCsd|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.rE)rArcr9rrr __getattr__s zInterchangeFrame.__getattr__r^cGs2|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.)r@select_columns_by_namerahasattrrF __class__re)rAr^framer9rrr simple_selects   zInterchangeFrame.simple_selectexprscGs 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.rE)rArkr9rrr selectszInterchangeFrame.selectN)r<r r#r=)r#r)r#r)rHrIr#r )r#rM)r#rV)r#rY)r#r`)rcrIr#r)r^rIr#r)rkrIr#r)rrrrr(_versionrBrDrGrLrUrXpropertyr_rbrdrjrlrrrr r;Ks           r;objrTypeIs[DataFrameLike]cCs t|dS)Nr?)rg)rorrr supports_dataframe_interchanges rq)r!r"r#r )rorr#rp) __future__renumtypingrrrnarwhals._utilsrrrOrTpyarrowpatyping_extensionsrr rJr narwhals.dtypesr narwhals.stable.v1.typingr IntEnumrr:r;rqrrrr s     .S