q-PhddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z erddlmZdd lmZddlm Z Gd d e Zd S) ) annotations)Sequence) accumulate) TYPE_CHECKING) PolarsColumn)CopyNotAllowedError) DataFrame)Iterator)AnyceZdZdZdZddd,d Z d-d.dZed/dZd0dZ d0dZ d0dZ d1dZ d2dZ d3dZd4dZd5d!Zd6d$Zd7d8d)Zd9d+Zd%S):PolarsDataFramea9 A dataframe object backed by a Polars DataFrame. Parameters ---------- df The Polars DataFrame backing the dataframe object. allow_copy Allow data to be copied during operations on this column. If set to `False`, a RuntimeError is raised if data would be copied. rT allow_copydfr rboolreturnNonec"||_||_dSN)_df _allow_copy)selfrrs \/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/interchange/dataframe.py__init__zPolarsDataFrame.__init__!s%F nan_as_nullcT|rd}t|t|j|S)a[ Construct a new dataframe object, potentially changing the parameters. Parameters ---------- nan_as_null Overwrite null values in the data with `NaN`. .. warning:: This functionality has not been implemented and the parameter will be removed in a future version. Setting this to `True` will raise a `NotImplementedError`. allow_copy Allow memory to be copied to perform the conversion. If set to `False`, causes conversions that are not zero-copy to fail. zfunctionality for `nan_as_null` has not been implemented and the parameter will be removed in a future version Use the default `nan_as_null=False`.r)NotImplementedErrorr r)rrrmsgs r __dataframe__zPolarsDataFrame.__dataframe__%s;*  +;  &c** *txJ????rdict[str, Any]ciS)zThe metadata for the dataframe.rs rmetadatazPolarsDataFrame.metadataCs  rintc|jjS)z.Return the number of columns in the dataframe.)rwidthr$s r num_columnszPolarsDataFrame.num_columnsHs x~rc|jjS)z+Return the number of rows in the dataframe.)rheightr$s rnum_rowszPolarsDataFrame.num_rowsLs xrc6|jdS)aQ Return the number of chunks the dataframe consists of. It is possible for a Polars DataFrame to consist of columns with a varying number of chunks. This method returns the number of chunks of the first column. See Also -------- polars.dataframe.frame.DataFrame.n_chunks first)rn_chunksr$s r num_chunkszPolarsDataFrame.num_chunksPsx  )))r list[str]c|jjS)zReturn the column names.)rcolumnsr$s r column_nameszPolarsDataFrame.column_names^s xrircb|j|}t||jS)z Return the column at the indicated position. Parameters ---------- i Index of the column. r)r to_seriesrr)rr5ss r get_columnzPolarsDataFrame.get_columnbs/ H  q ! !A$*:;;;;rnamestrcb|j|}t||jS)z Return the column with the given name. Parameters ---------- name Name of the column. r)rr9rr)rr:r8s rget_column_by_namez"PolarsDataFrame.get_column_by_namens/ H   % %A$*:;;;;rIterator[PolarsColumn]c#rK|jD]}t||jVdS)z(Return an iterator yielding the columns.rN)r get_columnsrr)rcolumns rr@zPolarsDataFrame.get_columnszsPh**,, D DFv$2BCCC C C C C D Drindices Sequence[int]ct|tsd}t|t|tst|}t |jdd|f|jS)z Create a new dataframe by selecting a subset of columns by index. Parameters ---------- indices Column indices z`indices` is not a sequenceNr) isinstancer TypeErrorlistr rr)rrBrs rselect_columnszPolarsDataFrame.select_columnsss'8,, !/CC.. '4(( $7mmG HQQQZ '    rnames Sequence[str]ct|tsd}t|t|j||jS)z Create a new dataframe by selecting a subset of columns by name. Parameters ---------- names Column names. z`names` is not a sequencer)rErrFr rselectr)rrIrs rselect_columns_by_namez&PolarsDataFrame.select_columns_by_namesU%** !-CC..  HOOE " "'    rNr/ int | NoneIterator[PolarsDataFrame]c#K|}|}|||kr|D]}t||jVdS|dks ||zdkrd|d}t |||z}|D]b}t |}||z}||zdkr|dz }t d||z|D])} t|| | |zddf|jV*cdS)a{ Return an iterator yielding the chunks of the dataframe. Parameters ---------- n_chunks The number of chunks to return. Must be a multiple of the number of chunks in the dataframe. If set to `None` (default), returns all chunks. Notes ----- When the columns in the dataframe are chunked unevenly, or when `n_chunks` is higher than the number of chunks in the dataframe, a slice must be performed that is not on the chunk boundary. This will trigger some compute for columns that contain null values and boolean columns. NrrzI`n_chunks` must be a multiple of the number of chunks of this dataframe ())r0_get_chunks_from_col_chunksr r ValueErrorlenrange) rr/total_n_chunkschunkschunkrsubchunks_per_chunksizestepstarts r get_chunkszPolarsDataFrame.get_chunkssz"**1133  (n"<"< J J%e8HIIIIIII J J!mmN!:a!?!?1-111 S// !#+n"<   5zz22--22AID"1d-@&@$GGE)eedl2AAA56#'#3  rIterator[DataFrame]c#K|d}d|D}dgtt|z}t t |dz D]}|||dz\}}|j||ddf}td|dDs,|j sd}t|| }|VdS) z Return chunks of this dataframe according to the chunks of the first column. If columns are not all chunked identically, they will be rechunked like the first column. If copy is not allowed, this raises a RuntimeError. rc6g|]}|Sr#)r[).0rYs r z?PolarsDataFrame._get_chunks_from_col_chunks..s <<<uzz||<<z>PolarsDataFrame._get_chunks_from_col_chunks..s&==!qAv======rallz*unevenly chunked columns must be rechunked) r9r^rGrrVrUrrhr/rrrechunk) r col_chunks chunk_sizesstartsr5r]endrYrs rrSz+PolarsDataFrame._get_chunks_from_col_chunkss__Q''2244 <<<<< tJ{33444s6{{Q''  AAE *JE3HU3Y\*E==u~~e'<'<===== ('3FC-c222 KKKK  r)rr rrrr)FT)rrrrrr )rr!)rr&)rr1)r5r&rr)r:r;rr)rr>)rBrCrr )rIrJrr r)r/rNrrO)rr_)__name__ __module__ __qualname____doc__versionrr propertyr%r)r,r0r4r9r=r@rHrMr^rSr#rrr r sr  G<@&&&&&& "@@@@@<X * * * *     < < < < < < < <DDDD     (    $*****Xrr N) __future__rcollections.abcr itertoolsrtypingrpolars.interchange.columnrpolars.interchange.protocolrr InterchangeDataFramer r polarsr r#rrr|s""""""$$$$$$ 222222;;;;;;IIIIII!((((((      TTTTT*TTTTTr