Para una lista dada de tuplas, si varias tuplas en la lista tienen el mismo primer elemento de tupla, entre ellas seleccione solo la tupla con el último elemento máximo.
Por ejemplo:
sample_list = [(5,16,2),(5,10,3),(5,8,1),(21,24,1)] En la lista de sample_list anterior, dado que las primeras 3 tuplas tienen el primer elemento 5 similar, en este caso, entre ellas, solo se debe conservar la segunda tupla, ya que tiene el último elemento máximo => 3 .
Operación esperada:
op = [(5,10,3),(21,24,1)]Código:
op = [] for m in range(len(sample_list)): li = [sample_list[m]] for n in range(len(sample_list)): if(sample_list[m][0] == sample_list[n][0] and sample_list[m][2] != sample_list[n][2]): li.append(sample_list[n]) op.append(sorted(li,key=lambda dd:dd[2],reverse=True)[0]) print (list(set(op)))Esto funciona. Pero es muy lento para una lista larga. ¿Hay una forma más pitónica o eficiente de hacer esto?
Utiliceitertools.groupby y operator.itemgetter para mejorar la legibilidad. Dentro de los grupos, aplique max con una función de tecla apropiada, nuevamente usando itemgetter por brevedad:
from itertools import groupby from operator import itemgetter as ig lst = [(5, 10, 3), (21, 24, 1), (5, 8, 1), (5, 16, 2)] [max(g, key=ig(-1)) for _, g in groupby(sorted(lst), key=ig(0))] # [(5, 10, 3), (21, 24, 1)] Para una solución de tiempo lineal, con espacio extra solo limitado por el número de primeros elementos únicos, puede usar un dict :
d = {} for tpl in lst: first, *_, last = tpl if first not in d or last > d[first][-1]: d[first] = tpl [*d.values()] # [(5, 10, 3), (21, 24, 1)]Prueba itertools.groupby :
from itertools import groupby sample_list.sort() print([max(l, key=lambda x: x[-1]) for _, l in groupby(sample_list, key=lambda x: x[0])]) O también con operator.itemgetter :
from itertools import groupby from operator import itemgetter sample_list.sort() print([max(l, key=itemgetter(-1)) for _, l in groupby(sample_list, key=itemgetter(0))])Para el rendimiento intente:
from operator import itemgetter dct = {} for i in sample_list: if i[0] in dct: dct[i[0]].append(i) else: dct[i[0]] = [i] print([max(v, key=itemgetter(-1)) for v in dct.values()])Toda la salida:
[(5, 10, 3), (21, 24, 1)]