Python中如何实现与集合类型无关的通用元素添加操作

Python 标准库没有提供面向所有可变容器的统一单元素添加接口，但可以通过符合鸭子类型设计的轻量适配完美实现需求，不需要硬编码类型判断，完全兼容list、set以及所有实现了对应添加方法的自定义容器。

方案1：通用单元素添加辅助函数 #

写一个极小的适配函数，自动识别容器支持的添加方法，完全遵循鸭子类型原则，不做强制类型判断：

def add(container, item):
    # 优先匹配set类容器的add方法
    if hasattr(container, "add"):
        return container.add(item)
    # 其次匹配list、deque类序列容器的append方法
    if hasattr(container, "append"):
        return container.append(item)
    raise TypeError(f"当前类型{type(container).__name__}不支持单元素添加操作")

把原示例代码里的col.append(x)替换为add(col, x)即可，直接支持list、set以及collections模块中的deque等绝大多数内置/第三方容器类型。

修改后的collate函数运行效果：

# list场景正常运行
>>> list(collate(range(6), 3))
[[0, 1, 2], [3, 4, 5]]
# set场景正常返回预期结果
>>> list(collate(range(6), 4, f_make=set))
[{0, 1, 2, 3}, {4, 5}]

方案2：支持自定义添加逻辑的参数注入 #

如果需要兼容更多特殊容器（比如dict、第三方实现的特殊集合类），可以把添加方法也做成可配置参数，和工厂函数配套，适配灵活性更高：

def collate(xs, n, f_make=list, f_add=None):
    if n < 1:
        raise ValueError("n < 1")
    col = f_make()
    # 未传入添加方法时自动识别
    if f_add is None:
        col_type = type(col)
        if hasattr(col, "add"):
            f_add = col_type.add
        elif hasattr(col, "append"):
            f_add = col_type.append
        else:
            raise TypeError(f"容器类型{col_type.__name__}无可用单元素添加方法，请手动传入f_add参数")
    for x in xs:
        if len(col) == n:
            yield col
            col = f_make()
        f_add(col, x)
    yield col

遇到特殊容器时，只需要传入对应f_make和f_add即可，不需要修改collate函数本身的逻辑。

注意事项 #

• 不要用isinstance()做类型判断来分支调用append/add，这种写法扩展性极差，无法兼容自定义容器、第三方库实现的集合类型，违背了鸭子类型的设计初衷。
• 原示例中f_make=lambda: list()可以简化为f_make=list，直接传类型本身作为工厂函数即可，代码更简洁，set场景同理直接传f_make=set。

Python标准库之所以没有统一单元素添加的方法名，本质是不同容器的添加语义存在核心差异：list.append是保留插入顺序、允许重复值的尾部追加操作，set.add是自动去重、不保证插入顺序的集合成员添加操作，语义差别决定了没有必要强行抽成统一接口，几行适配代码就足够覆盖所有通用场景。

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