如何根据任意条件函数过滤字典？

Question

我有一本要点词典，说：

>>> points={'a':(3,4), 'b':(1,2), 'c':(5,5), 'd':(3,3)}

我想创建一个新字典，其中包含所有 x 和 y 值小于 5 的点，即点 'a'、'b' 和 'd'。

根据书，每个字典都有

items()

函数，它返回一个

(key, pair)

元组列表：

>>> points.items()
[('a', (3, 4)), ('c', (5, 5)), ('b', (1, 2)), ('d', (3, 3))]

所以我写了这个：

>>> for item in [i for i in points.items() if i[1][0]<5 and i[1][1]<5]:
...     points_small[item[0]]=item[1]
...
>>> points_small
{'a': (3, 4), 'b': (1, 2), 'd': (3, 3)}

还有更优雅的方式吗？我期待 Python 有一些超级棒的

dictionary.filter(f)

函数...

Answer 1

您可以使用字典理解：

{k: v for k, v in points.items() if v[0] < 5 and v[1] < 5}

在 Python 2 中，从 2.7 开始：

{k: v for k, v in points.iteritems() if v[0] < 5 and v[1] < 5}

Answer 2

dict((k, v) for k, v in points.items() if all(x < 5 for x in v))

如果您使用的是 Python 2 并且

.iteritems()

可能有

lot

条目，您可以选择调用

.items()

而不是 points。

如果您确定每个点始终只是二维的，

all(x < 5 for x in v)

可能有点过分了（在这种情况下，您可能会用

and

表达相同的约束），但它会正常工作；-)。

Answer 3

points_small = dict(filter(lambda (a,(b,c)): b<5 and c < 5, points.items()))

Answer 4

>>> points = {'a': (3, 4), 'c': (5, 5), 'b': (1, 2), 'd': (3, 3)}
>>> dict(filter(lambda x: (x[1][0], x[1][1]) < (5, 5), points.items()))

{'a': (3, 4), 'b': (1, 2), 'd': (3, 3)}

Answer 5

dict((k, v) for (k, v) in points.iteritems() if v[0] < 5 and v[1] < 5)

Answer 6

我认为 Alex Martelli 的答案绝对是最优雅的方法，但只是想添加一种方法来满足您对超级棒的

dictionary.filter(f)

方法的需求，以 Python 的方式：

class FilterDict(dict):
    def __init__(self, input_dict):
        for key, value in input_dict.iteritems():
            self[key] = value
    def filter(self, criteria):
        for key, value in self.items():
            if (criteria(value)):
                self.pop(key)

my_dict = FilterDict( {'a':(3,4), 'b':(1,2), 'c':(5,5), 'd':(3,3)} )
my_dict.filter(lambda x: x[0] < 5 and x[1] < 5)

基本上我们创建一个继承自

dict

的类，但添加了过滤器方法。我们确实需要使用

.items()

进行过滤，因为在破坏性迭代时使用

.iteritems()

会引发异常。

Answer 7

dict((k, v) for (k, v) in points.iteritems() if v[0] < 5 and v[1] < 5)

Answer 8

不循环的递归：

def filter_age_city_recursive(list_dict, index=0, ag30=None):
   # Initialize the result list in the first call
   if ag30 is None:
      ag30 = []

   # Base case: If the index is equal to the length of the list, return the 
   #result
   if index == len(list_dict):
      return ag30
   # Recursive case: Process the current dictionary
   item = list_dict[index]
   if isinstance(item, dict):
       # Default age to 0 if not found
       age = item.get('age', 0)
       # Default city to empty string if not found
       city = item.get('city', '')
       if age > 30 and city == 'New York':
          ag30.append(item)

   # Recursive call to process the next dictionary in the list
   return filter_age_city_recursive(list_dict, index + 1, ag30)
# example
filtered_results = filter_age_city_recursive(data)
print(filtered_results)