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

我有一个点的字典，比如说:

>>> 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)函数…

288634 次浏览
小开 
dict((k, v) for (k, v) in points.iteritems() if v[0] < 5 and v[1] < 5)
小开 
dict((k, v) for (k, v) in points.iteritems() if v[0] < 5 and v[1] < 5)
小开 
points_small = dict(filter(lambda (a,(b,c)): b<5 and c < 5, points.items()))
小开 
dict((k, v) for k, v in points.items() if all(x < 5 for x in v))

如果你在Python 2中，你可以选择调用.iteritems()而不是.items()，并且points可能有很多的条目。

如果你确定每个点都是2D的，那么all(x < 5 for x in v)可能是多余的(在这种情况下，你可以用and来表达相同的约束)，但它会工作得很好;-)。

小开
最佳答案

你可以使用字典理解:

{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}
小开

我认为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的类，但添加了filter方法。我们确实需要使用.items()进行筛选，因为在破坏性迭代时使用.iteritems()将引发异常。

小开 
>>> 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)}

提交答案