我想知道是否可以實作 pandas api 訪問器（如https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.api.extensions.register_dataframe_accessor.html#pandas.api.extensions。 register_dataframe_accessor ) 用于 GroupBy 物件。
使用以下代碼，我可以將訪問器應用于組專案：
import pandas as pd
import numpy as np
@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()))
if __name__ == "__main__":
ds = pd.DataFrame({"longitude": np.linspace(0, 10),
"latitude": np.linspace(0, 20)})
ds['grp'] = ds['longitude'].astype(int)
for g in ds.groupby(by='grp'):
print(g[1].geo.center)
這導致
(0.40816326530612246, 0.8163265306122449)
(1.4285714285714286, 2.857142857142857)
(2.4489795918367347, 4.8979591836734695)
(3.4693877551020407, 6.938775510204081)
(4.4897959183673475, 8.979591836734695)
(5.510204081632653, 11.020408163265307)
(6.530612244897959, 13.061224489795919)
(7.551020408163266, 15.102040816326532)
(8.571428571428573, 17.142857142857146)
(9.489795918367347, 18.979591836734695)
(10.0, 20.0)
現在，我如何使用類似于以下內容的語法直接執行此操作：
ds.groupby('grp').geo.center
我為此得到的錯誤訊息是
ds.groupby(by='grp').geo.center
Traceback (most recent call last):
File "C:\.../ipykernel_11200/2937951017.py", line 1, in <module>
ds.groupby(by='grp').geo.center
File "C:\...\anaconda3\lib\site-packages\pandas\core\groupby\groupby.py", line 911, in __getattr__
raise AttributeError(
AttributeError: 'DataFrameGroupBy' object has no attribute 'geo'
uj5u.com熱心網友回復：
你本質上想做ds.groupby('grp').apply (lambda d: d.geo.center)嗎？
可以將其實作為訪問器，但您必須借用 Pandas 中 CachedAccessor 和 _register_accessor 的源代碼，然后定義您的訪問器物件，然后使用 _register_accessor 將其添加到 groupby 類中。以此為例。https://github.com/staircase-dev/piso/blob/master/piso/accessor.py
您的訪問器物件將參考它所附加到的 Groupby 物件。您想要定義center僅回傳.apply(lambda d: d.geo.center)Groupby 物件上的結果的屬性。不過，對于似乎是語法糖的東西來說，這需要做很多作業。
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/349181.html