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

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

如今,在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))

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

all(x < 5 for x in v)如果你确定每个点总是只有2D(在这种情况下你可能用a表示相同的约束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

我认为,亚历克斯·马尔泰利的答案肯定是最优雅的方式来做到这一点,但只是想补充的方式来满足你渴望得到一个超级真棒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()在破坏性迭代时使用会引发异常.

Answer 7

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