如何在Pandas DataFrame子类中覆盖loc、iloc、at访问器？

当然可以实现！虽然官方文档主要聚焦于添加新访问器，但覆盖loc、iloc、at这类内置访问器完全可行，核心思路就是在子类里重写对应的属性或包装访问器实例，先执行你的合理性检查，再衔接父类的原有逻辑。

下面分两种场景给你具体实现方案：

场景1：无需依赖访问参数的前置检查 #

如果你的检查逻辑不依赖用户具体要访问的行/列标签（比如检查索引是否唯一、DataFrame是否为空等），直接用@property重写访问器属性即可：

import pandas as pd

class CustomDataFrame(pd.DataFrame):
    @property
    def loc(self):
        # 第一步：执行自定义合理性检查
        print("触发loc访问前的全局检查...")
        if not self.index.is_unique:
            raise ValueError("自定义检查失败：DataFrame索引必须唯一！")
        # 第二步：调用父类的loc访问器，保留原有功能
        return super().loc
    
    # 同理重写iloc和at
    @property
    def iloc(self):
        print("触发iloc访问前的全局检查...")
        # 比如检查DataFrame是否为空
        if self.empty:
            raise ValueError("自定义检查失败：无法访问空DataFrame！")
        return super().iloc
    
    @property
    def at(self):
        print("触发at访问前的全局检查...")
        # 自定义检查逻辑示例：只允许访问特定列
        allowed_cols = ['name', 'age']
        if not all(col in allowed_cols for col in self.columns):
            raise PermissionError("自定义检查失败：存在未授权的列！")
        return super().at

测试一下：

# 创建索引不唯一的实例，调用loc会触发检查报错
df = CustomDataFrame({'a': [1,2]}, index=[0,0])
df.loc[0]  # 会抛出ValueError：自定义检查失败：DataFrame索引必须唯一！

场景2：需要依赖访问参数的针对性检查 #

如果你的检查需要根据用户实际传入的访问key（比如禁止访问某列、限制行范围），就需要包装Pandas内置的索引器类，重写其调用方法：

from pandas.core.indexing import _LocIndexer, _IlocIndexer, _AtIndexer

# 自定义Loc索引器，重写__getitem__来拦截访问参数
class CustomLocIndexer(_LocIndexer):
    def __getitem__(self, key):
        # 解析用户传入的访问key，做针对性检查
        print(f"正在检查访问参数：{key}")
        # 示例：禁止访问名为'forbidden_col'的列
        if isinstance(key, tuple) and 'forbidden_col' in key[1]:
            raise PermissionError("自定义检查失败：禁止访问forbidden_col列！")
        # 执行父类的原有逻辑
        return super().__getitem__(key)

# 同理可以包装iloc和at的索引器
class CustomIlocIndexer(_IlocIndexer):
    def __getitem__(self, key):
        # 示例：禁止访问最后一行
        if isinstance(key, tuple) and key[0] == -1:
            raise ValueError("自定义检查失败：禁止访问最后一行！")
        return super().__getitem__(key)

class CustomDataFrame(pd.DataFrame):
    @property
    def loc(self):
        # 返回自定义的索引器实例，传入当前DataFrame
        return CustomLocIndexer(self)
    
    @property
    def iloc(self):
        return CustomIlocIndexer(self)

测试一下：

df = CustomDataFrame({'name': ['Alice', 'Bob'], 'forbidden_col': [100, 200]})
df.loc[:, 'forbidden_col']  # 会抛出PermissionError：自定义检查失败：禁止访问forbidden_col列！
df.iloc[-1, :]  # 会抛出ValueError：自定义检查失败：禁止访问最后一行！

注意事项 #

• 尽量保留父类的所有原生功能，只添加必要的检查逻辑，避免破坏Pandas的原有行为；
• 如果你需要支持访问器的赋值操作（比如df.loc[0, 'a'] = 10），还要重写__setitem__方法，同理添加检查逻辑；
• 不同版本的Pandas中，索引器类的内部实现可能有细微差异，建议在对应版本的文档中确认索引器类的继承关系（比如_LocIndexer的位置）。

内容的提问来源于stack exchange，提问作者suyang