これを行うには、Pythonに組み込みが必要だと思います。アイテムのリストを取得し、キーを共通のキーを持つアイテムのリストにマッピングするディクショナリに変換します。
それはとても簡単です:
# using defaultdict
lookup = collections.defaultdict(list)
for item in items:
lookup[key(item)].append(item)
# or, using plain dict
lookup = {}
for item in items:
lookup.setdefault(key(item), []).append(item)
しかし、これは十分に頻繁に使用されるため、組み込み関数が適しています。次のように、自分で実装できます。
def grouped(iterable, key):
result = {}
for item in iterable:
result.setdefault(key(item), []).append(item)
return result
lookup = grouped(items, key)
これはitertools.groupby
、いくつかの重要な点で異なります。から同じ結果を得るにはgroupby
、これを行う必要がありますが、これは少し醜いです:
lookup = dict((k, list(v)) for k, v in groupby(sorted(items, key=key), key))
いくつかの例:
>>> items = range(10)
>>> grouped(items, lambda x: x % 2)
{0: [0, 2, 4, 6, 8], 1: [1, 3, 5, 7, 9]}
>>> items = 'hello stack overflow how are you'.split()
>>> grouped(items, len)
{8: ['overflow'], 3: ['how', 'are', 'you'], 5: ['hello', 'stack']}
より良い方法はありますか?