pandas.api.extensions.register_dataframe_accessor

pandas.api.extensions.register_dataframe_accessor(name) [source]

Регистрация пользовательского аксессора для объектов DataFrame.

Параметры:	name : str Имя, под которым аксессор должен быть зарегистрирован. Если имя конфликтует с уже существующим атрибутом, выводится предупреждение.

См. также

register_series_accessor, register_index_accessor

Примечания

При обращении к вашему аксессору он будет инициализирован объектом pandas, с которым пользователь взаимодействует. Таким образом, сигнатура должна быть

def __init__(self, pandas_object):  # noqa: E999
    ...

Для согласованности с методами pandas следует поднимать исключение AttributeError , если данные, переданные вашему аксессору, имеют неправильный тип данных.

>>> pd.Series(['a', 'b']).dt
Traceback (most recent call last):
...
AttributeError: Can only use .dt accessor with datetimelike values

Примеры

В вашем коде библиотеки:

import pandas as pd

@pd.api.extensions.register_dataframe_accessor("geo")
class GeoAccessor(object):
    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

Возвращаясь к интерактивной сессии IPython:

>>> ds = pd.DataFrame({'longitude': np.linspace(0, 10),
...                    'latitude': np.linspace(0, 20)})
>>> ds.geo.center
(5.0, 10.0)
>>> ds.geo.plot()
# plots data on a map