web-dev-qa-db-ja.com

任意の条件関数に従って辞書をフィルタリングする方法は?

ポイントの辞書があります。

>>> 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)関数を持つことを期待していました...

173
Adam Matan

現在、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}
364
Thomas
dict((k, v) for k, v in points.items() if all(x < 5 for x in v))

Python 2にいて、pointslotのエントリがある場合は、.iteritems()の代わりに.items()を呼び出すことを選択できます。

各ポイントが常に2Dのみであることがわかっている場合はall(x < 5 for x in v)が過剰になる可能性があります(その場合は、andを使用して同じ制約を表現できます)が、正常に機能します;-)。

106
Alex Martelli
points_small = dict(filter(lambda (a,(b,c)): b<5 and c < 5, points.items()))
20
sizzzzlerz
dict((k, v) for (k, v) in points.iteritems() if v[0] < 5 and v[1] < 5)
9
nosklo
>>> 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)}
9

Alex Martelliの答えは間違いなくこれを行う最もエレガントな方法だと思いますが、Pythonのような方法で超素晴らしいdictionary.filter(f)メソッドのあなたの欲求を満たす方法を追加したかっただけです:

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()を使用すると例外が発生するためです。

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