U Mfx@sdZddlmZddlmZmZddlZddlm Z m Z m Z ddl m Z ddlmZmZddlmZdd lmZmZed d d ZGd d d eZGdddeZdddddZdS)zc Base class for the internal managers. Both BlockManager and ArrayManager inherit from this class. ) annotations)TypeVarfinalN) ArrayLikeDtypeObjShapeAbstractMethodError)find_common_typenp_can_hold_element) PandasObject)Index default_indexT DataManager)boundc @s*eZdZUded<eddddZedddd Zeddd d Zed dd dZ eddddddZ d2ddddddddddZ ed3dddddddddZ dddddd Z ed!dd"d#d$Zd4dd%ddd&d'd(Zeddd)d*d+Zddd,d-Zddd)d.d/Zddd0d1ZdS)5rz list[Index]axesr returncCs t|dSNrselfr>/tmp/pip-unpacked-wheel-eb6vo0j3/pandas/core/internals/base.pyitems)szDataManager.itemsintcCs t|jSr)lenrrrrr__len__-szDataManager.__len__cCs t|jSr)rrrrrrndim1szDataManager.ndimrcCstdd|jDS)Ncss|]}t|VqdSr)r).0Zaxrrr 7sz$DataManager.shape..)tuplerrrrrshape5szDataManager.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_axis9szDataManager._validate_set_axisNFTrbool)rr$ allow_dupscopy consolidate only_slicerc Cs t|dSrr) rZnew_axisindexerr$ fill_valuer*r+r,r-rrrreindex_indexerJs zDataManager.reindex_indexer)r new_indexr$r,r-rc Cs,|j||\}}|j||||d||dS)z4 Conform data manager to new index. F)r$r/r+r,r-)rZreindexr0)rr1r$r/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_valuesoszDataManager._equal_valuesobject)r3rcCsVt|tsdS|j|j}}t|t|kr0dStddt||DsLdS||S)z5 Implementation for DataFrame.equals Fcss|]\}}||VqdSr)equals)rZax1Zax2rrrr sz%DataManager.equals..) isinstancerrrallzipr4)rr3Z self_axesZ other_axesrrrr6vs 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|Srrrrrrr,szDataManager.consolidatecCsdSrrrrrr_consolidate_inplacesz DataManager._consolidate_inplace)NFTTF)NTF)NF)__name__ __module__ __qualname____annotations__propertyrrrrr"r(r0r2r4r6r>r@rAr,rBrrrrr#sB   c@sXeZdZdZeeddddZddddZdd d d d Ze dddddZ dS)SingleDataManagerr&rrcCs |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)rIr7npZndarrayr 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. )rIrrtype from_array)rr?r;rLresindexZmgrrrrgrouped_reduces  z SingleDataManager.grouped_reducer )rLrQcCs t|dSrr)clsrLrQrrrrOszSingleDataManager.from_arrayN)F) rCrDrErrrGrIrMrR classmethodrOrrrrrHs rHzlist[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 )rUrrrinterleaved_dtypes rV)__doc__ __future__rtypingrrZnumpyrJZpandas._typingrrrZ pandas.errorsr Zpandas.core.dtypes.castr r Zpandas.core.baser Zpandas.core.indexes.apir rrrrHrVrrrrs    }2