q-PhddlmZddlmZddlmZmZmZmZddl m Z ddl m Z m Z mZmZmZddlmZerddlmZddlmZdd lmZdd l mZmZmZGd d e Zd S)) annotations) TYPE_CHECKING)Boolean CategoricalEnumString) PolarsBuffer)ColumnColumnNullTypeCopyNotAllowedError DtypeKind Endianness)polars_dtype_to_dtype)Iterator)Any)Series)CategoricalDescription ColumnBuffersDtypeceZdZdZddd&d Zd'd Zed'd Zed(dZed)dZ ed*dZ ed'dZ ed+dZ d'dZ d,d-dZd.dZd/d!Zd0d$Zd0d%ZdS)1 PolarsColumna6 A column object backed by a Polars Series. Parameters ---------- column The Polars Series backing the column object. allow_copy Allow data to be copied during operations on this column. If set to `False`, a RuntimeError will be raised if data would be copied. T allow_copycolumnrrboolreturnNonec"||_||_dSN)_col _allow_copy)selfrrs Y/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/interchange/column.py__init__zPolarsColumn.__init__%s %intc4|jS)zSize of the column in elements.)r lenr"s r#sizezPolarsColumn.size)sy}}r%cn|jjtkr|jdSdS)zOOffset of the first element with respect to the start of the underlying buffer.r)r dtyper_get_buffer_infor)s r#offsetzPolarsColumn.offset-s1 9?g % %9--//2 21r%rc8|jj}t|S)zData type of the column.)r r-r)r"pl_dtypes r#r-zPolarsColumn.dtype5s9?$X...r%rc|jj}|tkr*|jj}|jdk}n&|t kr |j}d}nd}t||dt||j dS)z Description of the categorical data type of the column. Raises ------ TypeError If the data type of the column is not categorical. physicalTz8`describe_categorical` only works on categorical columnsr) is_ordered is_dictionary categories) r r-rcatget_categoriesorderingrr6 TypeErrorrr!)r"r-r6r4msgs r#describe_categoricalz!PolarsColumn.describe_categorical;s  K  5577J:5JJ d]])JJJLCC.. %!&zd>NOOO   r%!tuple[ColumnNullType, int | None]cP|jdkrtjdfStjdfS)z7Description of the null representation the column uses.rN) null_countr NON_NULLABLE USE_BITMASKr)s r# describe_nullzPolarsColumn.describe_nullVs- ?a  !.4 4!-q0 0r%c4|jS)zThe number of null elements.)r r?r)s r#r?zPolarsColumn.null_count^sy##%%%r%dict[str, Any]ciS)zThe metadata for the column.r)s r#metadatazPolarsColumn.metadatacs  r%c4|jS)z3Return the number of chunks the column consists of.)r n_chunksr)s r# num_chunkszPolarsColumn.num_chunkshsy!!###r%NrI int | NoneIterator[PolarsColumn]c#K|}|j}|||kr|D]}t||jVdS|dks ||zdkrd|d}t |||z}|D]^}t |}||z}||zdkr|dz }td||z|D]%} t|| | |z|jV&_dS)a Return an iterator yielding the column chunks. Parameters ---------- n_chunks The number of chunks to return. Must be a multiple of the number of chunks in the column. Notes ----- When `n_chunks` is higher than the number of chunks in the column, a slice must be performed that is not on the chunk boundary. This will trigger some compute if the column contains null values or if the column is of data type boolean. NrrzF`n_chunks` must be a multiple of the number of chunks of this column ()r,)rJr get_chunksrr! ValueErrorr(range) r"rItotal_n_chunkschunkschunkr;subchunks_per_chunkr*stepstarts r#rOzPolarsColumn.get_chunkslsr"**%%''  (n"<"< G G"5T5EFFFFFFF G G!mmN!:a!?!?'#''' S// !#+n"<   5zz22--22AID"1d-@&@$GGE&eedl23@P   r%rc|jj}|tkr|jsd}t ||t krg|jjsI|jsd|jjd}t ||jj |_|j }| |d| |d| |ddS)z6Return a dictionary containing the underlying buffers.z string buffers must be convertedzcolumn z) must be converted to a local categoricalvaluesvalidityoffsets)datarZr[)r r-rr!r rr7is_localnameto_local _get_buffers_wrap_data_buffer_wrap_validity_buffer_wrap_offsets_buffer)r"r-r;bufferss r# get_bufferszPolarsColumn.get_bufferss  F??4#3?4C%c** * k ! !$)-*@*@*B*B !# /[ [[[)#... ..00DI)((****78+<==227:3FGG001CDD   r%buffertuple[PolarsBuffer, Dtype]c^t||j}t|j}||fS)Nr)r r!rr-r"rfinterchange_bufferr-s r#razPolarsColumn._wrap_data_buffers2)&T=MNNN%fl33!5((r% Series | None!tuple[PolarsBuffer, Dtype] | Nonecr|dSt||j}tjddtjf}||fS)Nrr,b)r r!r BOOLrNATIVEris r#rbz"PolarsColumn._wrap_validity_buffersB >4)&T=MNNNC):;!5((r%cr|dSt||j}tjddtjf}||fS)Nr@l)r r!r INTrrpris r#rcz!PolarsColumn._wrap_offsets_buffersB >4)&T=MNNNC):;!5((r%)rrrrrr)rr&)rr)rr)rr=)rrDr)rIrKrrL)rr)rfrrrg)rfrkrrl)__name__ __module__ __qualname____doc__r$r*propertyr/r-r<rBr?rGrJrOrerarbrcrFr%r#rrs  >B&&&&&&X///X/    X 4111X1&&&X&X$$$$)))))V    *)))) ))))))))))r%rN) __future__rtypingrpolars.datatypesrrrrpolars.interchange.bufferr polars.interchange.protocolr r r r rpolars.interchange.utilsrcollections.abcrrpolarsrrrrrrFr%r#rsT"""""" ????????????222222;:::::Y((((((XXXXXXXXXXk)k)k)k)k)6k)k)k)k)k)r%