U
|
¬ý°d ã @ sê d Z ddlmZ ddlmZmZmZ ddlZddl m
|
Z
|
m Z ddl m Z mZ ddlmZ ddlmZ erdd lmZ dd
|
lmZ ed d d Zeddd Zd dddddddZd ddddddddZG d d deZG d d d eZ dS ) z?
|
Shared methods for Index subclasses backed by ExtensionArray.
|
é )Ú annotations)Ú TYPE_CHECKINGÚCallableÚTypeVarN)Ú ArrayLikeÚnpt)Úcache_readonlyÚdoc)Ú ABCDataFrame)ÚIndex)Ú IntervalArray)ÚNDArrayBackedExtensionArrayÚ_TÚNDArrayBackedExtensionIndex)ÚboundÚ_ExtensionIndexTÚExtensionIndexFÚstrÚtypeÚbool)ÚnameÚdelegateÚcacheÚwrapc s² t | t ts"t jdkr|rJfdd}|_ j|_t|}q®fdd}ddfdd }|_ j|_t||}n*t s }n fd
|
d }|_ j|_|S ) aÚ
|
Make an alias for a method of the underlying ExtensionArray.
|
|
Parameters
|
----------
|
name : str
|
Name of an attribute the class should inherit from its EA parent.
|
delegate : class
|
cache : bool, default False
|
Whether to convert wrapped properties into cache_readonly
|
wrap : bool, default False
|
Whether to wrap the inherited result in an Index.
|
|
Returns
|
-------
|
attribute, method, property, or cache_readonly
|
Zgetset_descriptorc s t | j S ©N)ÚgetattrÚ_data©Úself©r © úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/indexes/extension.pyÚcached= s z"_inherit_from_data.<locals>.cachedc sZ t | j }rVt|t| jr4t| j|| jdS t|trH| | ¡S t|| jdS |S )Nr ) r r Ú
|
isinstancer Ú _simple_newr r
|
Ú set_indexr ©r Úresult)r r r r! ÚfgetF s
|
|
z _inherit_from_data.<locals>.fgetÚNone©Úreturnc s t | j | d S r )Úsetattrr ©r Úvaluer r r! ÚfsetP s z _inherit_from_data.<locals>.fsetc s| d|kr t dt| j | jf||}rxt|t| jrVt| j|| jdS t|trj| | ¡S t || jdS |S )NZinplacezcannot use inplace with r )
|
Ú
|
ValueErrorr Ú__name__r r# r$ r r
|
r% r )r ÚargsÚkwargsr' )Úattrr r r! Úmethod_ s
|
|
z"_inherit_from_data.<locals>.method)r r# Úpropertyr r1 Ú__doc__r Úcallable)r r r r r" r5 r( r/ r )r4 r r r! Ú_inherit_from_data# s$
|
|
|
r9 z list[str]z:Callable[[type[_ExtensionIndexT]], type[_ExtensionIndexT]])Únamesr r r r+ c s ddd fdd }|S )a
|
Class decorator to pin attributes from an ExtensionArray to a Index subclass.
|
|
Parameters
|
----------
|
names : List[str]
|
delegate : class
|
cache : bool, default False
|
wrap : bool, default False
|
Whether to wrap the inherited result in an Index.
|
ztype[_ExtensionIndexT])Úclsr+ c s* D ] }t | d}t| || q| S )N)r r )r9 r, )r; r Úmeth©r r r: r r r! Úwrapper s z inherit_names.<locals>.wrapperr )r: r r r r> r r= r! Ú inherit_namesq s r? c @ sH e Zd ZU dZded< dd Zeejd ddZe d d
|
d d Z
|
dS )r z>
|
Index subclass for indexes backed by ExtensionArray.
|
z+IntervalArray | NDArrayBackedExtensionArrayr c C s | j |¡S )zE
|
Convert value to be insertable to underlying array.
|
)r Z_validate_setitem_valuer- r r r! Ú_validate_fill_value s z#ExtensionIndex._validate_fill_valueNc C s^ z4|| }t |tjr t|}t |ts0td|W S tk
|
rX | t¡ |¡ Y S X d S )Nz,The map function must return an Index object) r# ÚnpZndarrayr Ú TypeErrorÚ ExceptionZastypeÚobjectÚmap)r ZmapperZ na_actionr' r r r! rE s
|
zExtensionIndex.mapznpt.NDArray[np.bool_]r* c C s
|
| j ¡ S r )r Zisnar r r r! Ú_isnan s zExtensionIndex._isnan)N) r1 Ú
|
__module__Ú __qualname__r7 Ú__annotations__r@ r r rE r rF r r r r! r s
|
c @ s8 e Zd ZU dZded< ddddZddd d
|
d Zd S ) r zK
|
Index subclass for indexes backed by NDArrayBackedExtensionArray.
|
r r z
|
np.ndarrayr* c C s | j jS r )r Ú_ndarrayr r r r! Ú_get_engine_target» s z.NDArrayBackedExtensionIndex._get_engine_targetr )r' r+ c C s |j | jjj kst| j |¡S r )Zdtyper rJ ÚAssertionErrorZ_from_backing_datar& r r r! Ú_from_join_target¾ s z-NDArrayBackedExtensionIndex._from_join_targetN)r1 rG rH r7 rI rK rM r r r r! r ´ s
|
)FF)FF) r7 Ú
|
__future__r Útypingr r r ÚnumpyrA Zpandas._typingr r Zpandas.util._decoratorsr r Zpandas.core.dtypes.genericr
|
Zpandas.core.indexes.baser Zpandas.core.arraysr Zpandas.core.arrays._mixinsr r r r9 r? r r r r r r! Ú<module> s( ÿO ÿ*
|
