วิธีกรองพจนานุกรมตามฟังก์ชั่นเงื่อนไขตามอำเภอใจ?

ฉันมีพจนานุกรมของจุดพูดว่า:

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

ฉันต้องการสร้างพจนานุกรมใหม่ด้วยคะแนนทั้งหมดที่ค่า x และ y น้อยกว่า 5 นั่นคือคะแนน 'a', 'b' และ 'd'

ตามพจนานุกรมหนังสือแต่ละเล่มมีitems()ฟังก์ชั่นซึ่งจะคืนค่ารายการของ(key, pair) tuple:

>>> 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)ฟังก์ชั่นสุดยอดเยี่ยม...

ทุกวันนี้ใน Python 2.7 ขึ้นไปคุณสามารถใช้ dict comprehension

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

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)อาจ overkill ถ้าคุณรู้แน่นอนว่าแต่ละจุดจะเป็นแบบ 2 มิติเท่านั้น (ในกรณีนั้นคุณอาจแสดงข้อ จำกัด เดียวกันกับand) แต่มันจะทำงานได้ดี ;-)

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)

ฉันคิดว่าคำตอบของ 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แต่เพิ่มวิธีการกรอง เราจำเป็นต้องใช้.items()สำหรับการกรองเนื่องจากการใช้.iteritems()ในขณะที่วนซ้ำแบบทำลายล้างจะทำให้เกิดข้อยกเว้น

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