U
|
¬ý°do ã @ s¼ d Z ddlmZ ddlmZmZmZ ddlZddl m
|
Z
|
m Z m Z m Z ddlmZ ddlmZmZ ddlmZ dd lmZmZ ed
|
d d ZG d d d eZG dd deZdddddZdS )zc
|
Base class for the internal managers. Both BlockManager and ArrayManager
|
inherit from this class.
|
é )Ú annotations)ÚLiteralÚTypeVarÚfinalN)Ú ArrayLikeÚAxisIntÚDtypeObjÚShape©ÚAbstractMethodError)Úfind_common_typeÚnp_can_hold_element)Ú PandasObject)ÚIndexÚ default_indexÚTÚ DataManager)Úboundc @ s$ e Zd ZU ded< eddddZedddd Zeddd
|
d Zed dd dZ eddddddZ
|
d3dddddddddZ ed4dddddd d d Z dddd d d!Z ed"dd#d$d%Zd5dd&dd'd(d)Zeddd*d+d,Zddd-d.Zddd*d/d0Zddd1d2ZdS )6r z list[Index]Úaxesr ©Úreturnc C s t | d S ©Nr
|
©Úself© r úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/internals/base.pyÚitems* s zDataManager.itemsÚintc C s
|
t | jS r )Úlenr r r r r Ú__len__. s zDataManager.__len__c C s
|
t | jS r )r r r r r r Úndim2 s zDataManager.ndimr c C s t dd | jD S )Nc s s | ]}t |V qd S r )r )Ú.0Zaxr r r Ú <genexpr>8 s z$DataManager.shape.<locals>.<genexpr>)Útupler r r r r Úshape6 s zDataManager.shaper ÚNone)ÚaxisÚ
|
new_labelsr c C sP t | j| }t |}|dkr.t | jdkr.n ||krLtd| d| dd S )Né r z#Length mismatch: Expected axis has z elements, new values have z elements)r r r Ú
|
ValueError)r r& r' Zold_lenZnew_lenr r r Ú_validate_set_axis: s ÿz DataManager._validate_set_axisNFTr Úbool)r r& Ú
|
allow_dupsÚcopyÚ
|
only_slicer c C s t | d S r r
|
)r Znew_axisÚindexerr& Ú
|
fill_valuer, r- r. r r r Úreindex_indexerK s
|
zDataManager.reindex_indexer)r Ú new_indexr& r. r c C s* | j | |¡\}}| j||||d|dS )z4
|
Conform data manager to new index.
|
F)r& r0 r- r. )r Zreindexr1 )r r2 r& r0 r. r/ r r r Ú reindex_axisW s úzDataManager.reindex_axis)r Úotherr c C s t | dS )z
|
To be implemented by the subclasses. Only check the column values
|
assuming shape and indexes have already been checked.
|
Nr
|
)r r4 r r r Ú _equal_valuesm s zDataManager._equal_valuesÚobject)r4 r c C sV t |tsdS | j|j }}t|t|kr0dS tdd t||D sLdS | |¡S )z5
|
Implementation for DataFrame.equals
|
Fc s s | ]\}}| |¡V qd S r )Úequals)r! Zax1Zax2r r r r" s z%DataManager.equals.<locals>.<genexpr>)Ú
|
isinstancer r r ÚallÚzipr5 )r r4 Z self_axesZ
|
other_axesr r r r7 t s
|
zDataManager.equalszlist[str] | None)r Ú
|
align_keysr c K s t | d S r r
|
)r Úfr; Úkwargsr r r Úapply s zDataManager.apply)r r c C s | j d|dS )Nr> )Úfunc)r> )r r? r r r Úisna s zDataManager.isnac C s dS )NTr r r r r Úis_consolidated s zDataManager.is_consolidatedc C s | S r r r r r r Ú consolidate s zDataManager.consolidatec C s d S r r r r r r Ú_consolidate_inplace s z DataManager._consolidate_inplace)NFTF)NF)N)Ú__name__Ú
|
__module__Ú __qualname__Ú__annotations__Úpropertyr r r r r$ r* r1 r3 r5 r7 r> r@ rA rB rC r r r r r % s<
|
ø û ýc @ s^ e Zd ZeddddZeeddddZddd d
|
Zd d Ze dd dddZ
|
dS )ÚSingleDataManagerz
|
Literal[1]r c C s dS )Nr( r r r r r r s zSingleDataManager.ndimr c C s
|
| j d S )zW
|
Quick access to the backing array of the Block or SingleArrayManager.
|
r )Zarraysr r r r Úarray¢ s zSingleDataManager.arrayr% c C s* | j }t|tjr t|j|}|||< dS )a'
|
Set values with indexer.
|
|
For Single[Block/Array]Manager, this backs s[indexer] = value
|
|
This is an inplace version of `setitem()`, mutating the manager/values
|
in place, not returning a new Manager (and Block), and thus never changing
|
the dtype.
|
N)rJ r8 ÚnpZndarrayr Zdtype)r r/ ÚvalueÚarrr r r Úsetitem_inplace« s
|
z!SingleDataManager.setitem_inplacec C s. | j }||}tt|}t| ||¡}|S r )rJ r r ÚtypeÚ
|
from_array)r r? rM ÚresÚindexZmgrr r r Úgrouped_reduce¿ s
|
z SingleDataManager.grouped_reducer )rM rR c C s t | d S r r
|
)ÚclsrM rR r r r rP Ç s z SingleDataManager.from_arrayN) rD rE rF rH r r rJ rN rS Ú classmethodrP r r r r rI s rI zlist[DtypeObj]zDtypeObj | None)Údtypesr c C s t | s dS t| S )zà
|
Find the common dtype for `blocks`.
|
|
Parameters
|
----------
|
blocks : List[DtypeObj]
|
|
Returns
|
-------
|
dtype : np.dtype, ExtensionDtype, or None
|
None is returned when `blocks` is empty.
|
N)r r )rV r r r Úinterleaved_dtypeÌ s rW ) Ú__doc__Ú
|
__future__r Útypingr r r ÚnumpyrK Zpandas._typingr r r r Z pandas.errorsr Zpandas.core.dtypes.castr r Zpandas.core.baser Zpandas.core.indexes.apir r r r rI rW r r r r Ú<module> s x/
|
