Làm thế nào để lọc một từ điển theo một hàm điều kiện tùy ý?

Tôi có một từ điển các điểm, nói:

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

Tôi muốn tạo một từ điển mới với tất cả các điểm có giá trị x và y nhỏ hơn 5, tức là các điểm 'a', 'b' và 'd'.

Theo cuốn sách , mỗi từ điển có items()chức năng, trả về một danh sách các (key, pair) tuple:

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

Vì vậy, tôi đã viết này:

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

Có cách nào thanh lịch hơn? Tôi đã mong đợi Python có một số dictionary.filter(f)chức năng siêu tuyệt vời ...

Ngày nay, trong Python 2.7 trở lên, bạn có thể sử dụng cách hiểu chính tả:

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

Và trong Python 3:

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

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

Bạn có thể chọn gọi .iteritems()thay vì .items()nếu bạn ở trong Python 2 và pointscó thể có nhiều mục.

all(x < 5 for x in v)có thể là quá mức nếu bạn biết chắc chắn mỗi điểm sẽ luôn chỉ là 2D (trong trường hợp đó bạn có thể biểu thị cùng một ràng buộc với một and) nhưng nó sẽ hoạt động tốt ;-).

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

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

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

Tôi nghĩ rằng câu trả lời của Alex Martelli chắc chắn là cách thanh lịch nhất để làm điều này, nhưng chỉ muốn thêm một cách để thỏa mãn mong muốn của bạn về một dictionary.filter(f)phương pháp siêu tuyệt vời theo cách của 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)

Về cơ bản chúng ta tạo một lớp kế thừa từ dict, nhưng thêm phương thức lọc. Chúng ta cần phải sử dụng .items()cho bộ lọc, vì sử dụng .iteritems()trong khi lặp đi lặp lại một cách triệt để sẽ tạo ra ngoại lệ.

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