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

我有一个点的字典，说：

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

我想创build一个新的字典，其所有点的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

如今，在Python 2.7及更高版本中，您可以使用dict理解：

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

在Python 3中：

 {k: v for k, v in points.items() 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中，你可以select调用.iteritems()而不是.items() ，并且points可能有很多条目。

如果你确实知道每个点总是只有2D（在这种情况下，你可能会用一个and表示相同的约束），但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

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

Answer 5

我认为，Alex Martelli的答案绝对是最优雅的方法，但只是想添加一种方法来满足你想要一个超级真棒dictionary.filter(f)方法在Pythonic的方式：

 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) 

基本上我们创build一个从dictinheritance的类，但添加了filter方法。 我们确实需要使用.items()进行过滤，因为使用.iteritems()而破坏性地迭代会引发exception。

Answer 6

 >>> 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 7

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