Comment puis-je regrouper efficacement une liste de paires en Python par le deuxième élément de chaque paire, tout en conservant le premier élément sous forme de liste dans le résultat groupé ?

Python Group By

On peut rencontrer le besoin de regrouper un ensemble de paires de données par le deuxième élément de chaque paire, tout en conservant le premier élément sous forme de liste au sein le résultat groupé. Cela peut être réalisé efficacement en Python en suivant les étapes suivantes.

Utilisation du dictionnaire

Créez un dictionnaire en utilisant le dict par défaut du module collections, où la clé est le deuxième élément de la paire. Ensuite, parcourez la liste d'entrée et ajoutez le premier élément à la valeur de la clé correspondante.

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

Convertissez le dictionnaire au format JSON attendu à l'aide d'une compréhension de liste :

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

Utilisation itertools.groupby

Une autre approche consiste à utiliser itertools.groupby mais nécessite que la liste d'entrée soit triée par le deuxième élément.

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

Créez une liste de dictionnaires en utilisant une compréhension :

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

Considérations sur la version Python

Avant Python 3.7, l'ordre des clés dans les dictionnaires n'est pas conservé. Pour conserver l'ordre d'origine, utilisez 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>

Depuis Python 3.7, les dictionnaires standards conservent l'ordre d'insertion, donc OrderedDict n'est plus nécessaire.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!