本文介绍如何设计一个通用的 python 集合类，使其在访问属性或调用方法时自动转发到内部每个元素，并支持链式操作与函数应用；同时说明其局限性及推荐实践方案。
本文介绍如何设计一个通用的 python 集合类，使其在访问属性或调用方法时自动转发到内部每个元素，并支持链式操作与函数应用；同时说明其局限性及推荐实践方案。
在 Python 中，我们常需要对一组同类对象（如自定义类 A 的多个实例）批量执行相同操作。理想情况下，希望像操作单个对象一样操作整个集合——例如 ac.some_attr 返回 [a1.some_attr, a2.some_attr]，ac.method() 返回 [a1.method(), a2.method()]，甚至支持 np.array(ac) 等外部函数的“透明”调用。这看似简单，实则涉及 Python 属性访问机制、协议方法（如 __array__）、以及外部库的类型检查逻辑。
✅ 可实现的部分：属性与方法的自动代理
通过重写 __getattr__，我们可以拦截所有未定义在集合类自身的属性/方法访问，并将其分发给每个元素：
from collections.abc import Sequence
class A_Collection(Sequence):
def __init__(self, items):
self._items = list(items)
def __getitem__(self, index):
return self._items[index]
def __len__(self):
return len(self._items)
def __getattr__(self, name):
# 获取每个元素的对应属性/方法
attrs = [getattr(item, name) for item in self._items]
# 若是可调用对象（方法），立即执行并收集结果
if attrs and callable(attrs[0]):
return [attr() for attr in attrs]
# 否则直接返回属性值列表
return attrs
def __repr__(self):
return f"{self.__class__.__name__}({self._items})"
使用示例：
class A:
def __init__(self, i): self.i = i
def add_one(self): return self.i + 1
ac = A_Collection([A(5), A(6), A(7)])
print(ac.i) # → [5, 6, 7]
print(ac.add_one()) # → [6, 7, 8]
该方案简洁有效，适用于大多数显式属性访问和方法调用场景。
⚠️ 不可绕过的限制：外部函数无法被自动拦截
关键点在于：Python 无法让第三方函数（如 np.array()、str()、json.dumps()）“感知”你的集合类并自动对其元素展开处理。这些函数通常依赖类型检查（如 isinstance(obj, np.ndarray)）或协议方法（如 __array__、__str__）。若你的集合类未显式实现这些协议，函数只会将其当作普通容器（如 list）或抛出 TypeError。
例如，np.array(ac) 默认会尝试将 ac 当作一个标量对象构造数组，而非调用 np.array(a) 对每个 a 执行——除非你主动实现 __array__：
def __array__(self, dtype=None):
# 显式支持 NumPy：对每个元素调用 np.array()
return np.array([np.array(item) for item in self._items], dtype=dtype)
类似地，若 A 支持 __array__，你也可统一委托：
def __array__(self, dtype=None):
return np.array([np.asarray(item, dtype=dtype) for item in self._items])
但注意：这要求你预先知道并手动适配每一个目标函数的协议接口，无法“泛化拦截未知函数”。
✅ 推荐方案：显式 apply() 方法 + 协议补全
更健壮、可维护的设计是结合两种策略：
- 提供
apply(func)方法，用于任意函数的批量应用； - 按需实现关键协议（如
__array__,__str__,__len__,__iter__）以兼容主流库。
完整推荐实现如下：
from collections.abc import Sequence
class A_Collection(Sequence):
def __init__(self, items):
self._items = list(items)
def __getitem__(self, i): return self._items[i]
def __len__(self): return len(self._items)
def apply(self, func, *args, **kwargs):
"""对每个元素应用 func，并返回新 A_Collection 实例"""
return type(self)(func(item, *args, **kwargs) for item in self._items)
def __getattr__(self, name):
attrs = [getattr(item, name) for item in self._items]
if attrs and callable(attrs[0]):
return [attr() for attr in attrs]
return attrs
# 显式支持 NumPy（关键！）
def __array__(self, dtype=None):
import numpy as np
return np.array([np.asarray(item, dtype=dtype) for item in self._items])
def __repr__(self):
return f"{self.__class__.__name__}({self._items})"
现在你可以：
ac = A_Collection([A(1), A(2)]) print(ac.apply(lambda x: x.i * 2)) # → A_Collection([2, 4]) print(np.array(ac)) # → array([1, 2]) （若 A 支持 __array__ 或 int 转换）
✅ 总结与最佳实践
- ✅
__getattr__是实现属性/方法自动代理的核心，适合“读取型”操作； - ❌ 不要试图让
np.array(ac)等外部函数“自动理解”你的集合——必须显式实现对应协议（如__array__）； - ✅
apply()是最灵活、最可控的批量操作入口，应作为首选 API； - ? 避免过度魔法：若集合行为复杂，优先考虑组合模式（如
map(func, ac)）或显式列表推导，保证代码可读性与调试友好性； - ? 如需生产级功能，可基于
collections.abc.MutableSequence扩展，并添加map,filter,reduce等高阶方法。
最终，一个优雅的集合类不是追求“完全透明”，而是提供清晰、一致、可预测的批量操作契约——apply() 即是这一契约最坚实的基础。
