U
|
¬ý°d ã @ sd d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z
|
erPd dl m Z m Z G dd de
|
Z d S )
|
é )Ú annotations)Úabc)Ú TYPE_CHECKING)Ú PandasColumn)Ú DataFrame)r ÚIndexc @ sØ e Zd ZdZd-ddddddd Zd.ddd d
|
d d Zed dddZddddZddddZ ddddZ
|
ddddZ ddd d d Z d dd d!d"Z d#dd$d%Zd dd&d'Zd dd(d)Zd/d+d,Zd*S )0ÚPandasDataFrameXchga
|
A data frame class, with only the methods required by the interchange
|
protocol defined.
|
Instances of this (private) class are returned from
|
``pd.DataFrame.__dataframe__`` as objects with the methods and
|
attributes defined on this class.
|
FTr ÚboolÚNone)ÚdfÚ nan_as_nullÚ
|
allow_copyÚreturnc C s || _ || _|| _dS )zz
|
Constructor - an instance of this (private) class is returned from
|
`pd.DataFrame.__dataframe__`.
|
N)Ú_dfÚ _nan_as_nullÚ _allow_copy)Úselfr r r © r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/interchange/dataframe.pyÚ__init__ s z PandasDataFrameXchg.__init__)r r r c C s t | j||S ©N)r r )r r r r r r Ú __dataframe__( s z!PandasDataFrameXchg.__dataframe__zdict[str, Index])r c C s d| j jiS )Nz pandas.index)r Úindex©r r r r Úmetadata- s z PandasDataFrameXchg.metadataÚintc C s t | jjS r )Úlenr Úcolumnsr r r r Ú num_columns3 s z PandasDataFrameXchg.num_columnsc C s
|
t | jS r )r r r r r r Únum_rows6 s z PandasDataFrameXchg.num_rowsc C s dS )Né r r r r r Ú
|
num_chunks9 s z PandasDataFrameXchg.num_chunksr c C s | j jS r ©r r r r r r Ú column_names< s z PandasDataFrameXchg.column_namesr )Úir c C s t | jjd d
|f | jdS ©N©r )r r Úilocr )r r$ r r r Ú
|
get_column? s z PandasDataFrameXchg.get_columnÚstr)Únamer c C s t | j| | jdS r% ©r r r )r r* r r r Úget_column_by_nameB s z&PandasDataFrameXchg.get_column_by_namezlist[PandasColumn]c s fdd j jD S )Nc s g | ]}t j| jd qS )r& r+ )Ú.0r* r r r Ú
|
<listcomp>F s ÿz3PandasDataFrameXchg.get_columns.<locals>.<listcomp>r" r r r r Ú get_columnsE s
|
þz PandasDataFrameXchg.get_columnsc C sF t |tjstdt |ts&t|}t| jjd d
|f | j| j S )Nz`indices` is not a sequence)
|
Ú
|
isinstancer ÚSequenceÚ
|
ValueErrorÚlistr r r' r r )r Úindicesr r r Úselect_columnsK s
|
ÿz"PandasDataFrameXchg.select_columnsc C sF t |tjstdt |ts&t|}t| jjd d
|f | j| j S )Nz`names` is not a sequence)
|
r0 r r1 r2 r3 r r Úlocr r )r Únamesr r r Úselect_columns_by_nameU s
|
ÿz*PandasDataFrameXchg.select_columns_by_nameNc c s~ |rt|dkrtt | j}|| }|| dkr2|d7 }td|| |D ].}t| jj|||
dd
f | j| jV qBn| V dS )z9
|
Return an iterator yielding the chunks.
|
r r N)r r Úranger r' r r )r Zn_chunksÚsizeÚstepÚstartr r r Ú
|
get_chunks_ s
|
ý
|
z PandasDataFrameXchg.get_chunks)FT)FT)N)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r Úpropertyr r r r! r# r( r, r/ r5 r8 r= r r r r r s& ÿ ÿ
|
|
r N)Ú
|
__future__r Ú collectionsr Útypingr Z pandas.core.interchange.columnr Z*pandas.core.interchange.dataframe_protocolr Z DataFrameXchgZpandasr r r r r r Ú<module> s
|
