如何重载Python列表的[]及推导式实例化以添加自定义map方法？

首先明确：没法直接重载Python里[]字面量或者列表推导式的实例化逻辑——这俩语法是硬编码绑定到原生list类的，Python没给开发者提供钩子去替换这个默认行为。

不过有几个实用的替代方案能帮你简化代码：

1. 给自定义类加转换方法 #

在你的自定义MyList类里加个类方法，专门用来把原生列表/生成器转成自定义实例：

class MyList(list):
    def map(self, func):
        # 直接返回自定义类实例，支持链式调用
        return MyList(map(func, self))
    
    @classmethod
    def from_iter(cls, iterable):
        return cls(iterable)

使用示例：

# 把[]创建的原生列表转成MyList
my_list = MyList.from_iter([1,2,3])
# 直接把列表推导式转成MyList
my_list = MyList.from_iter(x*2 for x in range(5))

2. 直接用自定义类包裹迭代器/列表推导式 #

其实不用额外方法，直接把列表推导式或者字面量传给MyList的构造器就行，和你用list()实例化的逻辑一致：

# 替代[]创建自定义列表
my_list = MyList([1,2,3])
# 替代列表推导式创建自定义列表
my_list = MyList(x for x in range(10) if x % 2 == 0)

3. 用自定义函数简化写法 #

如果觉得每次写MyList()麻烦，可以封装个短名字的函数：

class MyList(list):
    def map(self, func):
        return MyList(map(func, self))

def ml(iterable=None):
    if iterable is None:
        return MyList()
    return MyList(iterable)

使用时就像原生列表一样简洁：

my_list = ml([1,2,3])
my_list = ml(x+1 for x in [1,2,3])

不推荐的方案：替换全局/局部的list指向 #

你可以在局部作用域里把list赋值成你的自定义类：

list = MyList
# 现在用list()会创建MyList实例
my_list = list([1,2,3])

但这个做法风险很高，会破坏原生list的正常使用，尤其是导入依赖原生list的模块时，极易出现莫名其妙的bug，不建议这么做。

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