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

问题描述：

我有一本积分词典，说：

>>> 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)功能……

解决方案 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}

解决方案 2：

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

如果您使用的是 Python 2，并且可能有很多条目，那么您可以选择调用.iteritems()而不是。.items()`points`

all(x < 5 for x in v)如果您确信每个点始终都是 2D 的，那么这可能会有些过度（在这种情况下，您可以使用 来表达相同的约束and），但它可以正常工作；-)。

解决方案 3：

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

解决方案 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)}

解决方案 5：

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

解决方案 6：

我认为 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)

基本上，我们创建一个继承自的类dict，但添加了 filter 方法。我们确实需要使用.items()进行过滤，因为.iteritems()在破坏性迭代时使用会引发异常。

解决方案 7：

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