Python类构造函数中动态赋值__getitem__报错，如何解决？

问题 #

Python允许直接将函数赋值给实例成员，比如：

class A:
    def __init__(self):
        self.test = lambda x: print(x)

a = A()
a.test(10) # 输出10

但像__getitem__这类特殊方法，直接赋值给实例成员却行不通。假设我们需要根据flag变量为__getitem__提供两种实现，常规写法是在方法内部加if-else判断：

class A:
    def __init__(self, N, flag):
        self.values = list(range(N))
        self.flag = flag
        self.N = N
    def __getitem__(self, idx):
        if self.flag: 
            return self.values[idx]
        else:
            return self.values[idx] * self.N    # 示例实现

a = A(10, True)
a[5] # 返回5

但在面向数据的场景中，我们希望避免每次调用__getitem__都走if-else分支，想在构造函数里直接确定并赋值对应的实现，于是尝试了下面的写法：

class A:
    def __init__(self, N, flag):
        self.values = list(range(N))
        self.flag = flag
        self.N = N
        if flag: 
            self.__getitem__ = lambda self, idx: self.values[idx]
        else:
            self.__getitem__ = lambda self, idx: self.values[idx] * self.N

a = A(10, False)
a[5] # 预期返回50，实际抛出NotImplementedError

为什么这种写法会报错？有没有可行的实现方式？Python是否已经对这类if-else分支做了优化？

解决方案 #

为什么直接赋值实例的__getitem__不生效？ #

Python的特殊方法（以__开头和结尾的方法，比如__getitem__、__str__）是在类层面查找的，而非实例层面。当执行a[5]时，Python会去type(a)也就是A类中查找__getitem__，完全不会读取实例a的同名属性，所以你给实例赋值的方法根本不会被调用。

可行的实现方式 #

方式1：用实例方法代理（推荐）

在类中定义一个通用的__getitem__，让它调用我们在构造函数里赋值的实例自定义方法，这样既避免了重复判断，又不会影响其他实例：

class A:
    def __init__(self, N, flag):
        self.values = list(range(N))
        self.N = N
        # 给实例绑定具体的实现逻辑
        if flag:
            self._get_impl = lambda idx: self.values[idx]
        else:
            self._get_impl = lambda idx: self.values[idx] * self.N
    
    def __getitem__(self, idx):
        # 统一转发到实例绑定的实现
        return self._get_impl(idx)

a = A(10, False)
print(a[5]) # 输出50，符合预期

方式2：工厂函数生成不同子类

如果两种实现逻辑差异较大，可定义两个子类，用工厂函数根据flag返回对应实例，结构更清晰：

class BaseA:
    def __init__(self, N):
        self.values = list(range(N))
        self.N = N

class AFlagTrue(BaseA):
    def __getitem__(self, idx):
        return self.values[idx]

class AFlagFalse(BaseA):
    def __getitem__(self, idx):
        return self.values[idx] * self.N

def create_A(N, flag):
    return AFlagTrue(N) if flag else AFlagFalse(N)

a = create_A(10, False)
print(a[5]) # 输出50

方式3：动态修改类方法（不推荐）

直接修改类的__getitem__会影响所有后续创建的实例，容易引发意外问题，仅在特殊场景下使用：

class A:
    def __init__(self, N, flag):
        self.values = list(range(N))
        self.N = N
        # 修改类的方法，所有新实例都会用这个实现
        if flag:
            A.__getitem__ = lambda self, idx: self.values[idx]
        else:
            A.__getitem__ = lambda self, idx: self.values[idx] * self.N

关于if-else的优化 #

CPython等解释器会对简单分支判断做基础优化，多数场景下，__getitem__里的if-else开销可以忽略不计。只有在极端高频调用的场景下，才需要考虑上述避免分支的方案。

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