如何在Python中包装内置list并实现自定义filter方法？

想要定义一个ListWrapper类，使其拥有内置list的所有行为，同时新增一个filter方法。预期效果如下：

ls = [1, 2, 3, 4, 5]
ls = ListWrapper(ls)
ls.filter(lambda x: True if x % 2 == 0 else False)
print(ls)
# 输出: [2, 4]

以下是失败的尝试：

class ListWrapper(list):
    def filter(self, func):
        filtered_list = list(filter(func, self))
        self = ListWrapper(filtered_list)
ls = [1, 2, 3, 4, 5]
ls = ListWrapper(ls)
ls.filter(lambda x: True if x % 2 == 0 else False)
print(ls)
# 输出: [1, 2, 3, 4, 5]

你的代码失效的核心原因是：在filter方法内对self赋值，只是修改了方法内部的局部变量引用，并没有改变原实例本身的内容。要实现预期效果，有两种可行方案：

方案一：修改当前实例内容 #

直接修改原列表的内部元素，这样调用filter后原实例会被更新，符合你预期的调用方式：

class ListWrapper(list):
    def filter(self, func):
        # 直接替换当前列表的所有元素
        self[:] = list(filter(func, self))

ls = [1, 2, 3, 4, 5]
ls = ListWrapper(ls)
ls.filter(lambda x: x % 2 == 0)  # 简化lambda表达式写法
print(ls)
# 输出: [2, 4]

这里self[:] = ...是Python中修改列表内部元素的标准写法，它不会改变self的引用，只是更新列表里的元素。

方案二：返回新的ListWrapper实例 #

这种方式更贴合Python内置函数的设计风格（不修改原对象，返回新对象），但调用时需要重新赋值：

class ListWrapper(list):
    def filter(self, func):
        filtered_elements = list(filter(func, self))
        return ListWrapper(filtered_elements)

ls = [1, 2, 3, 4, 5]
ls = ListWrapper(ls)
ls = ls.filter(lambda x: x % 2 == 0)  # 必须重新赋值接收新实例
print(ls)
# 输出: [2, 4]

如果希望和你最初的调用代码完全一致（不需要重新赋值），方案一是更合适的选择。

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