o Ри╩h<'у@s▄dZddlmZddlmZmZddlZddlmZddl m Z GddДdГZGd d Дd ГZddd Дdfd+ddДZ GddДdГZedddНd,dd ДГZeed!d"Нd,d#d$ДГZeed%d"Нd,d&d'ДГZeed(d"Нd,d)d*ДГZdS)-zЖ accessor.py contains base classes for implementing accessor properties that can be mixed into or pinned onto other pandas classes. щ)┌annotations)┌Callable┌finalN)┌doc)┌find_stack_levelcsTeZdZUeГZded<eГZded<ed ddДГZ d dd ДZ dЗfddДZЗZS)┌ DirNamesMixin·set[str]┌ _accessorszfrozenset[str]┌ _hidden_attrs┌returncCs|j|jBS)z: Delete unwanted __dir__ for this object. )r r й┌selfйr·F/var/www/html/env/lib/python3.10/site-packages/pandas/core/accessor.py┌_dir_deletionsszDirNamesMixin._dir_deletionscsЗfddДИjDГS)z9 Add additional __dir__ for this object. csh|] }tИ|Гr|ТqSr)┌hasattr)┌.0┌accessorrrr┌ "sz/DirNamesMixin._dir_additions..)r rrrr┌_dir_additionsszDirNamesMixin._dir_additions· list[str]cs*ttГабГ}||аб|абB}t|ГS)zА Provide method name lookup and completion. Notes ----- Only provide 'public' methods. )┌set┌super┌__dir__rr┌sorted)r ┌rvй┌ __class__rrr$szDirNamesMixin.__dir__)rr)rr) ┌__name__┌ __module__┌__qualname__rr ┌__annotations__┌ frozensetr rrrr┌ __classcell__rrrrrs rc@sHeZdZdZdddДZdddДZddd ДZed ddДd fdddДГZdS)┌PandasDelegatez@ Abstract base class for delegating methods/properties. ┌name┌strcOєtd|ЫЭГВ)NzYou cannot access the property й┌ TypeErrorйr r%┌args┌kwargsrrr┌_delegate_property_get6єz%PandasDelegate._delegate_property_getcOstd|ЫdЭГВ)Nz The property z cannot be setr()r r%┌valuer+r,rrr┌_delegate_property_set9sz%PandasDelegate._delegate_property_setcOr')NzYou cannot call method r(r*rrr┌_delegate_method<r.zPandasDelegate._delegate_methodFcCє|SйNrй┌xrrr┌FєzPandasDelegate.T┌ accessorsr┌typ┌ overwrite┌bool┌accessor_mapping·Callable[[str], str]┌raise_on_missingr┌Nonecs|d ЗЗfddД}d ЗЗfddД}|D])} |s!tИИ| ГdГdur!q|dkr*|| Г} n|| Г} |s5t|| Гs;t|| | ГqdS) aэ Add accessors to cls from the delegate class. Parameters ---------- cls Class to add the methods/properties to. delegate Class to get methods/properties and doc-strings. accessors : list of str List of accessors to add. typ : {'property', 'method'} overwrite : bool, default False Overwrite the method/property in the target class if it exists. accessor_mapping: Callable, default lambda x: x Callable to map the delegate's function to the cls' function. raise_on_missing: bool, default True Raise if an accessor does not exist on delegate. False skips the missing accessor. r%r&cs>ЗfddД}ЗfddД}И|_И|_t||tИИИГГjdНS)Ncs |аИбSr3)r-rйr%rr┌_getter_s z[PandasDelegate._add_delegate_accessors.._create_delegator_property.._gettercs|аИ|бSr3)r0)r ┌ new_valuesr@rr┌_setterbsz[PandasDelegate._add_delegate_accessors.._create_delegator_property.._setter)┌fget┌fsetr)r┌property┌getattr┌__doc__)r%rArCйr<┌delegater@r┌_create_delegator_property^s¤zJPandasDelegate._add_delegate_accessors.._create_delegator_propertycs(ЗfddД}И|_tИИИГГj|_|S)Ncs|jИg|вRi|дОSr3)r1)r r+r,r@rr┌foszSPandasDelegate._add_delegate_accessors.._create_delegator_method..f)rrGrH)r%rLrIr@r┌_create_delegator_methodnszHPandasDelegate._add_delegate_accessors.._create_delegator_methodNrFйr%r&)rGr┌setattr)┌clsrJr8r9r:r<r>rKrMr%rLrrIr┌_add_delegate_accessors?s АЄz&PandasDelegate._add_delegate_accessorsNrN)r8rr9r&r:r;r<r=r>r;rr?) rrr rHr-r0r1┌classmethodrQrrrrr$1s ∙r$FcCr2r3rr4rrrr6Нr7r6Tr8rr9r&r:r;r<r=r>csЗЗЗЗЗЗfddД}|S)a╫ Add delegated names to a class using a class decorator. This provides an alternative usage to directly calling `_add_delegate_accessors` below a class definition. Parameters ---------- delegate : object The class to get methods/properties & doc-strings. accessors : Sequence[str] List of accessor to add. typ : {'property', 'method'} overwrite : bool, default False Overwrite the method/property in the target class if it exists. accessor_mapping: Callable, default lambda x: x Callable to map the delegate's function to the cls' function. raise_on_missing: bool, default True Raise if an accessor does not exist on delegate. False skips the missing accessor. Returns ------- callable A class decorator. Examples -------- @delegate_names(Categorical, ["categories", "ordered"], "property") class CategoricalAccessor(PandasDelegate): [...] cs|jИИИИИИdН|S)N)r:r<r>)rQ)rPйr<r8rJr:r>r9rr┌add_delegate_accessors░s·z.delegate_names..add_delegate_accessorsr)rJr8r9r:r<r>rTrrSr┌delegate_namesИs(rUc@s"eZdZdZdddДZdd ДZd S)┌CachedAccessoraб Custom property-like object. A descriptor for caching accessors. Parameters ---------- name : str Namespace that will be accessed under, e.g. ``df.foo``. accessor : cls Class with the extension methods. Notes ----- For accessor, The class's __init__ method assumes that one of ``Series``, ``DataFrame`` or ``Index`` as the single argument ``data``. r%r&rr?cCs||_||_dSr3)┌_name┌ _accessor)r r%rrrr┌__init__╪s zCachedAccessor.__init__cCs,|dur|jS|а|б}tа||j|б|Sr3)rX┌object┌__setattr__rW)r ┌objrP┌accessor_objrrr┌__get__▄s zCachedAccessor.__get__N)r%r&rr?)rrr rHrYr^rrrrrV─s rV┌)┌klass┌othersr%csЗЗfddД}|S)aG Register a custom accessor on {klass} objects. Parameters ---------- name : str Name under which the accessor should be registered. A warning is issued if this name conflicts with a preexisting attribute. Returns ------- callable A class decorator. See Also -------- register_dataframe_accessor : Register a custom accessor on DataFrame objects. register_series_accessor : Register a custom accessor on Series objects. register_index_accessor : Register a custom accessor on Index objects. Notes ----- When accessed, your accessor will be initialized with the pandas object the user is interacting with. So the signature must be .. code-block:: python def __init__(self, pandas_object): # noqa: E999 ... For consistency with pandas methods, you should raise an ``AttributeError`` if the data passed to your accessor has an incorrect dtype. >>> pd.Series(['a', 'b']).dt Traceback (most recent call last): ... AttributeError: Can only use .dt accessor with datetimelike values Examples -------- In your library code:: import pandas as pd @pd.api.extensions.register_dataframe_accessor("geo") class GeoAccessor: def __init__(self, pandas_obj): self._obj = pandas_obj @property def center(self): # return the geographic center point of this DataFrame lat = self._obj.latitude lon = self._obj.longitude return (float(lon.mean()), float(lat.mean())) def plot(self): # plot this array's data on a map, e.g., using Cartopy pass Back in an interactive IPython session: .. code-block:: ipython In [1]: ds = pd.DataFrame({{"longitude": np.linspace(0, 10), ...: "latitude": np.linspace(0, 20)}}) In [2]: ds.geo.center Out[2]: (5.0, 10.0) In [3]: ds.geo.plot() # plots data on a map cs^tИИГrtjdt|ГЫdtИГЫdtИГЫdЭttГdНtИИtИ|ГГИjа Иб|S)Nzregistration of accessor z under name z for type z: is overriding a preexisting attribute with the same name.)┌ stacklevel) r┌warnings┌warn┌repr┌UserWarningrrOrVr ┌add)rйrPr%rr┌ decorator2s √z%_register_accessor..decoratorr)r%rPrirrhr┌_register_accessorщsI rj┌ DataFrame)r`cCєddlm}t||ГS)Nr)rk)┌pandasrkrj)r%rkrrr┌register_dataframe_accessorBє rn┌SeriescCrl)Nr)rp)rmrprj)r%rprrr┌register_series_accessorIrorq┌IndexcCrl)Nr)rr)rmrrrj)r%rrrrr┌register_index_accessorPrors) r8rr9r&r:r;r<r=r>r;rN)rH┌ __future__r┌typingrrrc┌pandas.util._decoratorsr┌pandas.util._exceptionsrrr$rUrVrjrnrqrsrrrr┌s*[·< % X