U cP@sdZddlmZddlmZmZmZddlZddl m Z m Z m Z ddl mZddlmZmZddlmZdd lmZmZed d d ZGd d d eZGdddeZdddddZdS)zc Base class for the internal managers. Both BlockManager and ArrayManager inherit from this class. ) annotations)LiteralTypeVarfinalN) ArrayLikeDtypeObjShapeAbstractMethodError)find_common_typenp_can_hold_element) PandasObject)Index default_indexT DataManager)boundc@s&eZdZUded<eddddZedddd Zeddd d Zed dd dZ eddddddZ d2dddddddddZ ed3ddddddddZ dddddd Z ed!dd"d#d$Zd4dd%ddd&d'd(Zeddd)d*d+Zddd,d-Zddd)d.d/Zddd0d1ZdS)5rz list[Index]axesrreturncCs t|dSNr selfr>/tmp/pip-unpacked-wheel-g7fro6k3/pandas/core/internals/base.pyitems*szDataManager.itemsintcCs t|jSr)lenrrrrr__len__.szDataManager.__len__cCs t|jSr)rrrrrrndim2szDataManager.ndimrcCstdd|jDS)Ncss|]}t|VqdSr)r).0Zaxrrr 8sz$DataManager.shape..)tuplerrrrrshape6szDataManager.shapeNone)axis new_labelsrcCsPt|j|}t|}|dkr.t|jdkr.n||krLtd|d|ddS)Nrz#Length mismatch: Expected axis has z elements, new values have z elements)rrr ValueError)rr%r&Zold_lenZnew_lenrrr_validate_set_axis:szDataManager._validate_set_axisNFTrbool)rr% allow_dupscopy only_slicercCs t|dSrr )rZnew_axisindexerr% fill_valuer+r,r-rrrreindex_indexerKs zDataManager.reindex_indexer)r new_indexr%r-rcCs*|j||\}}|j||||d|dS)z4 Conform data manager to new index. F)r%r/r,r-)rZreindexr0)rr1r%r/r-r.rrr reindex_axisWs zDataManager.reindex_axis)rotherrcCs t|dS)z To be implemented by the subclasses. Only check the column values assuming shape and indexes have already been checked. Nr )rr3rrr _equal_valuesmszDataManager._equal_valuesobject)r3rcCsVt|tsdS|j|j}}t|t|kr0dStddt||DsLdS||S)z5 Implementation for DataFrame.equals Fcss|]\}}||VqdSr)equals)r Zax1Zax2rrrr!sz%DataManager.equals..) isinstancerrrallzipr4)rr3Z self_axesZ other_axesrrrr6ts zDataManager.equalszlist[str] | None)r align_keysignore_failuresrcKs t|dSrr )rfr:r;kwargsrrrapplyszDataManager.apply)rrcCs|jd|dS)Nr>)func)r>)rr?rrrisnaszDataManager.isnacCsdS)NTrrrrris_consolidatedszDataManager.is_consolidatedcCs|Srrrrrr consolidateszDataManager.consolidatecCsdSrrrrrr_consolidate_inplacesz DataManager._consolidate_inplace)NFTF)NF)NF)__name__ __module__ __qualname____annotations__propertyrrrrr#r)r0r2r4r6r>r@rArBrCrrrrr$s>   c@sfeZdZeddddZeeddddZddd d Zdd d ddZe dddddZ dS)SingleDataManagerz Literal[1]rcCsdS)Nr'rrrrrrszSingleDataManager.ndimrcCs |jdS)zW Quick access to the backing array of the Block or SingleArrayManager. r)ZarraysrrrrarrayszSingleDataManager.arrayr$cCs*|j}t|tjrt|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)rJr7npZndarrayr Zdtype)rr.valuearrrrrsetitem_inplaces   z!SingleDataManager.setitem_inplaceFr*)r;cCs.|j}||}tt|}t|||}|S)z ignore_failures : bool, default False Not used; for compatibility with ArrayManager/BlockManager. )rJrrtype from_array)rr?r;rMresindexZmgrrrrgrouped_reduces  z SingleDataManager.grouped_reducer)rMrRcCs t|dSrr )clsrMrRrrrrPszSingleDataManager.from_arrayN)F) rDrErFrHrrrJrNrS classmethodrPrrrrrIs rIzlist[DtypeObj]zDtypeObj | None)dtypesrcCst|s dSt|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)rr )rVrrrinterleaved_dtypes rW)__doc__ __future__rtypingrrrZnumpyrKZpandas._typingrrrZ pandas.errorsr Zpandas.core.dtypes.castr r Zpandas.core.baser Zpandas.core.indexes.apirrrrrIrWrrrrs    z4