Tengo una lista de tuplas:

start_list = [(A99, 2, 3 B1), (A21, 3, 4, B1), (A123, 4, 5, B2), (A22, 3 6, B2), (A12, 4, 6, B1)]

Y quiero agrupar todas las tuplas con el mismo último elemento en listas de la forma:

result = [[B1, [A99, A21, A12]],[B2, [A123, A22]]]

Mi intento:

for idx in range(len(start_list)):
        concat_list += [start_list[idx][0]]
        for idx2 in range(idx+1, len(start_list)):
            if start_list[idx][-1] == start_list[idx2][-1]:
                concat_list += [start_list[idx2][0]]

        grouped_list.append([start_list[idx][-1], concat_list])
        concat_list = []

    grouped_list = dict(((x[0]), x) for x in grouped_list).values()

No tiene en cuenta todas las tuplas, tampoco incluye todos los elementos iniciales (A99, A123 ...) en la tupla correspondiente.

0
Qubix 16 oct. 2018 a las 13:48

2 respuestas

La mejor respuesta

Puede utilizar collections.defaultdict para una O ( n ) solución:

from collections import defaultdict

start_list = [('A99', 2, 3, 'B1'), ('A21', 3, 4, 'B1'), ('A123', 4, 5, 'B2'),
              ('A22', 3, 6, 'B2'), ('A12', 4, 6, 'B1')]

res = defaultdict(list)

for value, _, _, key in start_list:
    res[key].append(value)

Resultado:

defaultdict(list, {'B1': ['A99', 'A21', 'A12'],
                   'B2': ['A123', 'A22']})

Si no le importan los elementos tuple, para la estructura anidada puede usar:

res_lst = list(res.items())

[('B1', ['A99', 'A21', 'A12']), ('B2', ['A123', 'A22'])]

O para su salida precisa deseada, use una lista de comprensión:

res_lst = [[k, v] for k, v in res.items()]

[['B1', ['A99', 'A21', 'A12']], ['B2', ['A123', 'A22']]]
2
jpp 16 oct. 2018 a las 10:59

Podrías probar algo así:

from collections import defaultdict

d = defaultdict(lambda: [])
start_list = [('A99', 2, 3, 'B1'), ('A21', 3, 4, 'B1'), ('A123', 4, 5, 'B2'), ('A22', 3, 6, 'B2'), ('A12', 4, 6, 'B1')]

for elt in start_list:
    d[elt[3]].append(elt[0])

grouped_list = [[k, v] for k, v in d.items()]
0
William Labaguette 16 oct. 2018 a las 11:00