如何通过pandas的register_dataframe_accessor修改原始DataFrame？

问题核心：在自定义Accessor中直接替换self._df的引用（比如self._df = 新DataFrame）只会改变Accessor内部的指向，不会影响外部的df变量——因为外部变量仍指向最初创建的空DataFrame对象。

以下是几种实用的解决方式：

1. 原地修改原始DataFrame的内容 #

不替换整个DataFrame对象，而是直接修改其内部数据，这样外部变量能同步看到变化：

import pandas as pd

@pd.api.extensions.register_dataframe_accessor("geo")
class GeoAccessor:
    def __init__(self, pandas_obj):
        self._df = pandas_obj

    def ingest_sample_data(self):
        # 构造示例地理数据
        sample_df = pd.DataFrame({
            "latitude": [39.9042, 31.2304, 23.1291],
            "longitude": [116.4074, 121.4737, 113.2644]
        })
        
        # 清空原DataFrame并写入新数据
        self._df.drop(self._df.columns, axis=1, inplace=True)
        self._df[sample_df.columns] = sample_df

调用后原始df会被更新：

df = pd.DataFrame()
df.geo.ingest_sample_data()
print(df)
# 输出包含latitude和longitude列的DataFrame

2. 返回新DataFrame让用户重新赋值 #

遵循pandas函数式风格，避免原地修改的副作用，让方法返回新的DataFrame，用户手动更新变量：

@pd.api.extensions.register_dataframe_accessor("geo")
class GeoAccessor:
    def __init__(self, pandas_obj):
        self._df = pandas_obj

    def ingest_sample_data(self):
        sample_df = pd.DataFrame({
            "latitude": [39.9042, 31.2304, 23.1291],
            "longitude": [116.4074, 121.4737, 113.2644]
        })
        return sample_df

调用方式：

df = pd.DataFrame()
df = df.geo.ingest_sample_data()
print(df)

3. 提供inplace参数（推荐） #

模仿pandas原生API的设计，添加inplace选项，让用户自由选择原地修改或返回新对象：

@pd.api.extensions.register_dataframe_accessor("geo")
class GeoAccessor:
    def __init__(self, pandas_obj):
        self._df = pandas_obj

    def ingest_sample_data(self, inplace=False):
        sample_df = pd.DataFrame({
            "latitude": [39.9042, 31.2304, 23.1291],
            "longitude": [116.4074, 121.4737, 113.2644]
        })
        
        if inplace:
            self._df.drop(self._df.columns, axis=1, inplace=True)
            self._df[sample_df.columns] = sample_df
            return None
        else:
            return sample_df

调用方式：

# 原地修改
df = pd.DataFrame()
df.geo.ingest_sample_data(inplace=True)

# 或返回新对象
df = pd.DataFrame()
df = df.geo.ingest_sample_data()

内容的提问来源于stack exchange，提问作者Pietro D'Antuono