如何在Python自定义类中实现推导式（Comprehension）

好问题！咱们一步步来实现你想要的自定义类“推导式”功能，分两种场景讨论：序列类（MySequence）和映射类（MyMapping）的推导式风格创建方式。

一、实现MySequence的推导式风格创建 #

首先得搞清楚：元组所谓的“推导式”tuple(i for i in range(3))本质上并不是特殊语法，只是把生成器表达式直接传给了tuple的构造函数——而tuple的构造函数本身就支持接收任意可迭代对象。

所以咱们的MySequence只要在构造函数里处理可迭代参数就行，代码示例：

class MySequence:
    def __init__(self, iterable=None):
        # 用列表存储内部元素，支持接收任意可迭代对象
        self.items = list(iterable) if iterable is not None else []
    
    def __repr__(self):
        # 自定义打印格式，方便直观查看结果
        return f"MySequence({', '.join(map(repr, self.items))})"

# 测试效果
>>> MySequence(i for i in range(3))
MySequence(0, 1, 2)
>>> MySequence(x**2 for x in [1,2,3])
MySequence(1, 4, 9)
>>> MySequence("abc")
MySequence('a', 'b', 'c')

这样就完美实现了类似元组推导式的用法，生成器表达式、列表、字符串等任何可迭代对象都能直接传给MySequence创建实例。

二、实现MyMapping的“推导式”风格创建 #

这里得先说明一个限制：Python中{...}字面量是内置类型专属语法（比如dict、set），自定义类无法直接使用MyMapping{str(i): i for i in range(3)}这种写法——这是语法层面硬编码的，没有开放给自定义类的扩展接口。

不过我们可以实现等价的推导式创建效果，有两种实用方案：

方案1：让构造函数支持接收字典推导式/键值对可迭代对象 #

和dict的逻辑对齐，让MyMapping的构造函数既能接收字典，也能接收键值对组成的生成器表达式：

class MyMapping:
    def __init__(self, source=None):
        self.data = {}
        if source is not None:
            # 兼容两种输入：字典，或者(key, value)形式的可迭代对象
            if isinstance(source, dict):
                self.data.update(source)
            else:
                try:
                    self.data.update(source)
                except TypeError:
                    raise ValueError("MyMapping需要传入字典或键值对可迭代对象")
    
    def __repr__(self):
        items = ', '.join(f"{repr(k)}: {repr(v)}" for k, v in self.data.items())
        return f"MyMapping({{{items}}})"

# 测试效果
>>> MyMapping({str(i): i for i in range(3)})
MyMapping({'0': 0, '1': 1, '2': 2})
# 也可以直接传键值对生成器，和dict的用法完全一致
>>> MyMapping((str(i), i*2) for i in range(3))
MyMapping({'0': 0, '1': 2, '2': 4})

方案2：自定义类方法强化“推导式”语义 #

如果觉得直接传推导式不够直观，可以专门写一个类方法来明确语义：

class MyMapping:
    def __init__(self, data=None):
        self.data = dict(data) if data is not None else {}
    
    @classmethod
    def from_comprehension(cls, key_value_gen):
        # 专门接收推导式风格的键值对生成器
        return cls(key_value_gen)
    
    def __repr__(self):
        items = ', '.join(f"{repr(k)}: {repr(v)}" for k, v in self.data.items())
        return f"MyMapping({{{items}}})"

# 测试效果
>>> MyMapping.from_comprehension((str(i), i+1) for i in range(3))
MyMapping({'0': 1, '1': 2, '2': 3})

总结 #

• 序列类的推导式效果：只要让构造函数支持接收可迭代对象（包括生成器表达式）就能轻松实现，完全贴合Python内置类型的用法；
• 映射类的推导式效果：因为{...}语法无法扩展，我们可以通过构造函数接收字典推导式或键值对生成器来达到等价效果，符合Python“显式优于隐式”的设计哲学。

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