Python字典Filter类仅返回首个符合条件元素的原因探究

我正在为字典实现过滤功能，编写了如下Filter类的isin和gt_x方法：

class Filter:

  def __init__(self, obj):
    self.obj = obj

  def isin(self, key, lst):
    self.obj = {k:[v[i] for i in range(len(self.obj)+1) if self.obj[key][i] in lst] 
                        for k,v in self.obj.items()}    

  def gt_x(self,key,value):
    self.obj = {k:[v[i] for i in range(len(self.obj)+1) if self.obj[key][i] > value] 
                        for k,v in self.obj.items()} 

测试时发现，这两个方法仅返回首个符合过滤条件的元素，而非所有符合条件的元素。例如：

d = {'a':[1,1,2,2,3,3],
     'b':[3,3,2,2,1,1,]}

f = Filter(d)     
f.isin(key='a',lst=[2,3])
f.obj
>>> {'a': [2], 'b': [2]} # desired: {'a': [2,2,3,3], 'b': [2,2,1,1]}

以及：

d = {'a':[1,1,2,2,3,3],
     'b':[3,3,2,2,1,1,]}

f = Filter(d)
f.gt_x('a', value=1)
f.obj 
>>> {'a': [2], 'b': [2]} # desired: {'a': [2,2,3,3], 'b': [2,2,1,1]}

我清楚实际可借助Pandas等库实现该功能，但出于好奇，想了解为何当前代码仅返回首个符合条件的元素？

原因分析 #

问题源于两个关键错误：

1. 循环范围错误：你使用range(len(self.obj)+1)作为索引范围，但len(self.obj)是字典的键的数量（测试案例中为2），所以实际只遍历了索引0、1、2，而不是列表的全部6个元素。

2. 推导式中修改原对象导致的截断：在字典推导式执行过程中，你直接对self.obj进行赋值。当处理第一个键（如'a'）时，self.obj已经被替换为新字典，此时新字典中'a'的列表只有1个元素。后续处理'b'键时，self.obj[key][i]的索引范围被限制在这个新列表的长度内，最终只保留了第一个符合条件的元素。

修正方案 #

先提前计算所有符合条件的索引，再基于这些索引构建新字典，避免在推导式中修改原对象：

class Filter:

  def __init__(self, obj):
    self.obj = obj

  def isin(self, key, lst):
    # 预计算所有有效索引
    valid_indices = [i for i, val in enumerate(self.obj[key]) if val in lst]
    # 基于索引生成过滤后的字典
    self.obj = {k: [v[i] for i in valid_indices] for k, v in self.obj.items()}    

  def gt_x(self, key, value):
    valid_indices = [i for i, val in enumerate(self.obj[key]) if val > value]
    self.obj = {k: [v[i] for i in valid_indices] for k, v in self.obj.items()} 

测试修正后的代码：

d = {'a':[1,1,2,2,3,3],
     'b':[3,3,2,2,1,1,]}

f = Filter(d)     
f.isin(key='a',lst=[2,3])
print(f.obj)
# 输出: {'a': [2, 2, 3, 3], 'b': [2, 2, 1, 1]}

f2 = Filter(d.copy())
f2.gt_x('a', value=1)
print(f2.obj)
# 输出: {'a': [2, 2, 3, 3], 'b': [2, 2, 1, 1]}

这样就能得到预期的过滤结果。

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