U f/e@sdZddlmZddlmZddlmZmZmZddl m Z ddl m Z ddl mZddlmZed d d ZGd d d eZGd ddeZdddddZdS)zc Base class for the internal managers. Both BlockManager and ArrayManager inherit from this class. ) annotations)TypeVar)DtypeObjShapefinalAbstractMethodError)find_common_type) PandasObject)IndexT DataManager)boundc @seZdZUded<eddddZdddd Zeddd d Zed dd dZe ddddddZ d,ddddddddddZ e d-dddddddddZ dddddd Z d!dd"d#d$Zd.dd%ddd&d'd(Zddd)d*d+ZdS)/r z list[Index]axesr )returncCs t|dSNrselfr>/tmp/pip-unpacked-wheel-tiezk1ph/pandas/core/internals/base.pyitemsszDataManager.itemsintcCs t|jSr)lenrrrrr__len__"szDataManager.__len__cCs t|jSr)rrrrrrndim%szDataManager.ndimrcCstdd|jDS)Ncss|]}t|VqdSr)r).0Zaxrrr +sz$DataManager.shape..)tuplerrrrrshape)szDataManager.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_axisNFTr bool)rr allow_dupscopy consolidate only_slicerc Cs t|dSrr) rZnew_axisindexerr fill_valuer&r'r(r)rrrreindex_indexer>s 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))rZreindexr,)rr-r r+r(r)r*rrr reindex_axisKs 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)rr/rrr _equal_valuescszDataManager._equal_valuesobject)r/rcCsVt|tsdS|j|j}}t|t|kr0dStddt||DsLdS||S)z5 Implementation for DataFrame.equals Fcss|]\}}||VqdSr)equals)rZax1Zax2rrrrtsz%DataManager.equals..) isinstancer rrallzipr0)rr/Z self_axesZ other_axesrrrr2js zDataManager.equalszlist[str] | None)r align_keysignore_failuresrcKs t|dSrr)rfr6r7kwargsrrrapplyyszDataManager.apply)rrcCs|jd|dS)Nr:)func)r:)rr;rrrisnaszDataManager.isna)NFTTF)NTF)NF)__name__ __module__ __qualname____annotations__propertyrrrrrr$r,r.r0r2r:r<rrrrr s6   c@seZdZdZeddZdS)SingleDataManagerr"cCs |jdS)zW Quick access to the backing array of the Block or SingleArrayManager. r)ZarraysrrrrarrayszSingleDataManager.arrayN)r=r>r?rrArCrrrrrBsrBzlist[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 )rDrrrinterleaved_dtypes rEN)__doc__ __future__rtypingrZpandas._typingrrrZ pandas.errorsrZpandas.core.dtypes.castr Zpandas.core.baser Zpandas.core.indexes.apir r r rBrErrrrs       n