Bagaimana cara memfilter kamus sesuai dengan fungsi kondisi arbitrer?

Saya memiliki kamus poin, katakan:

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

Saya ingin membuat kamus baru dengan semua titik yang nilainya x dan y lebih kecil dari 5, yaitu titik 'a', 'b' dan 'd'.

Menurut buku itu , setiap kamus memiliki items()fungsi, yang mengembalikan daftar (key, pair) tuple:

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

Jadi saya telah menulis ini:

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

Apakah ada cara yang lebih elegan? Saya mengharapkan Python untuk memiliki beberapa dictionary.filter(f)fungsi yang luar biasa ...

Saat ini, dalam Python 2.7 dan lebih tinggi, Anda dapat menggunakan pemahaman dict:

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

Dan dengan 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))

Anda dapat memilih untuk menelepon .iteritems()daripada .items()menggunakan Python 2 dan pointsmungkin memiliki banyak entri.

all(x < 5 for x in v)mungkin berlebihan jika Anda tahu pasti setiap titik akan selalu menjadi 2D saja (dalam hal ini Anda mungkin mengekspresikan kendala yang sama dengan a and) tetapi itu akan berfungsi dengan baik ;-).

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)

Saya pikir jawaban Alex Martelli jelas merupakan cara paling elegan untuk melakukan ini, tetapi hanya ingin menambahkan cara untuk memuaskan keinginan Anda akan dictionary.filter(f)metode super luar biasa dengan cara 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)

Pada dasarnya kami membuat kelas yang mewarisi dari dict, tetapi menambahkan metode filter. Kita perlu menggunakan .items()untuk penyaringan, karena menggunakan .iteritems()iterasi sementara secara destruktif akan menimbulkan pengecualian.

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