Python でペアのリストを各ペアの 2 番目の要素で効率的にグループ化し、グループ化された結果内の最初の要素をリストとして保持するにはどうすればよいでしょうか?

Python Group By

最初の要素をリストとして保持しながら、各ペアの 2 番目の要素でデータ ペアのセットをグループ化する必要がある場合があります。グループ化された結果。これは、次の手順を使用して Python で効率的に実現できます。

辞書の使用

コレクション モジュールのdefaultdictを使用して辞書を作成します。キーはペアの 2 番目の要素です。次に、入力リストを反復処理し、最初の要素を対応するキーの値に追加します。

<code class="python">import collections

input = [
    ('11013331', 'KAT'),
    ('9085267', 'NOT'),
    ('5238761', 'ETH'),
    ('5349618', 'ETH'),
    ('11788544', 'NOT'),
    ('962142', 'ETH'),
    ('7795297', 'ETH'),
    ('7341464', 'ETH'),
    ('9843236', 'KAT'),
    ('5594916', 'ETH'),
    ('1550003', 'ETH'),
]

res = collections.defaultdict(list)
for v, k in input:
    res[k].append(v)</code>

リスト内包表記を使用して辞書を期待される JSON 形式に変換します。

<code class="python">result = [{'type': k, 'items': v} for k, v in res.items()]</code>

使用itertools.groupby

別のアプローチでは itertools.groupby を使用しますが、入力リストを 2 番目の要素でソートする必要があります。

<code class="python">from operator import itemgetter
from itertools import groupby

sorted_input = sorted(input, key=itemgetter(1))
groups = groupby(sorted_input, key=itemgetter(1))</code>

内包表記を使用して辞書のリストを作成します。

<code class="python">result = [{'type': k, 'items': [x[0] for x in v]} for k, v in groups]</code>

Python バージョンに関する考慮事項

Python 3.7 より前では、辞書内のキーの順序は保持されません。元の順序を維持するには、collections.OrderedDict を使用します。

<code class="python">from collections import OrderedDict

res = OrderedDict()
for v, k in input:
    if k in res:
        res[k].append(v)
    else:
        res[k] = [v]</code>

Python 3.7 以降、通常の辞書は挿入順序を維持するため、OrderedDict は必要なくなりました。

以上がPython でペアのリストを各ペアの 2 番目の要素で効率的にグループ化し、グループ化された結果内の最初の要素をリストとして保持するにはどうすればよいでしょうか?の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。