本文介绍一种轻量级、高性能的 AttrView 类,用于创建对对象列表中指定属性的实时只读视图,使 random.choices() 能自动感知属性变更,避免重复构建权重列表。
本文介绍一种轻量级、高性能的 `attrview` 类,用于创建对对象列表中指定属性的实时只读视图,使 `random.choices()` 能自动感知属性变更,避免重复构建权重列表。
在使用 random.choices() 进行加权随机抽样时,若权重来源于对象的可变属性(如 champ.odds),直接预生成 list_odds = [c.odds for c in champs] 会导致权重与对象状态脱节——修改 champ.odds 后,原权重列表不会自动更新,必须手动同步,既易出错又违背 Python 的“显式优于隐式”原则。
理想方案是提供一个惰性、动态、只读的属性视图:它不复制数据,不缓存值,而是在每次访问(如索引或迭代)时实时读取对象当前属性值。这正是 AttrView 的设计目标。
以下是完整、生产就绪的 AttrView 实现(兼容 Python 3.12+,向下兼容 3.9+):
from collections.abc import Sequencefrom typing import overload, Any, TypeVar, GenericT = TypeVar('T')class AttrView(Generic[T], Sequence[T]): def __init__(self, seq: Sequence[Any], attr: str): self._seq = seq self._attr = attr @overload def __getitem__(self, index: int) -> T: ... @overload def __getitem__(self, index: slice) -> list[T]: ... def __getitem__(self, index): if isinstance(index, slice): return [getattr(item, self._attr) for item in self._seq[index]] return getattr(self._seq[index], self._attr) def __len__(self) -> int: return len(self._seq) def __repr__(self) -> str: return f"AttrView({self._seq!r}, {self._attr!r})"
✅ 关键特性说明:
? 使用示例:
import randomclass Champ: def __init__(self, odds: float): self.odds = odds def __repr__(self): return f"Champ(odds={self.odds})"# 初始化选手列表与动态权重视图champs = [Champ(0.25), Champ(0.5)]weights = AttrView[float](champs, "odds") # 显式类型注解提升可读性# 首次采样(按初始 odds)print("Initial draw:", random.choices(champs, weights=weights, k=5))# → 可能输出: [Champ(odds=0.25), Champ(odds=0.5), ...]# 动态更新某选手 oddschamps[0].odds = 0.99# 再次采样 —— weights 自动反映新值!无需任何额外操作print("After update:", random.choices(champs, weights=weights, k=5))# → 此时 champ0 被选中的概率显著升高
⚠️ 注意事项:
总结:AttrView 是一个简洁、符合 Python 哲学的通用工具——它用最少的代码解决了对象属性与权重解耦的核心痛点,兼顾性能、可维护性与可读性,推荐作为加权随机系统的基础组件复用。