Python: condicionalmente eliminar elementos de la lista

Question

Supongamos que tengo una lista de tuplas:

x = [(1,2), (3,4), (7,4), (5,4)] 

De todas las tuplas que comparten el segundo elemento, quiero preservar la tupla con el mayor primer elemento:

y = [(1,2), (7,4)] 

¿Cuál es la mejor manera de lograr esto en Python?

---

Gracias por las respuestas.

• Las tuplas podrían ser listas de dos elementos en su lugar, si eso hace la diferencia.
• Todos los elementos son enteros no negativos.
• Me gustan las respuestas actuales. Realmente debería aprender más sobre lo que collections tiene para ofrecer.

Answer 1

Similar a la respuesta de Aaron

>>> from collections import defaultdict 
>>> x = [(1,2), (3,4), (7,4), (5,4)] 
>>> d = defaultdict(int) 
>>> for v,k in x: 
... d[k] = max(d[k],v) 
... 
>>> y=[(k,v) for v,k in d.items()] 
>>> y 
[(1, 2), (7, 4)] 

nota que la orden no se conserva con este método. Para preservar la orden use esto en su lugar

>>> y = [(k,v) for k,v in x if d[v]==k] 
>>> y 
[(1, 2), (7, 4)] 

aquí está otra manera. Se utiliza más capacidad de almacenamiento, pero tiene menos llamadas a un máximo, por lo que puede ser más rápido

>>> d = defaultdict(list) 
>>> for k,v in x: 
... d[v].append(k) 
... 
>>> y = [(max(k),v) for v,k in d.items()] 
>>> y 
[(1, 2), (7, 4)] 

Una vez más, una simple modificación conserva el orden

>>> y = [(k,v) for k,v in x if max(d[v])==k] 
>>> y 
[(1, 2), (7, 4)] 

Answer 2

uso collections.defaultdict

import collections 

max_elements = collections.defaultdict(tuple) 

for item in x: 
    if item > max_elements[item[1]]: 
     max_elements[item[1]] = item 

y = max_elements.values() 

Answer 3

mi propio intento, un poco inspirado en aaronsterling:

(oh sí, todos los elementos son no negativos)

def processtuples(x): 
    d = {} 
    for item in x: 
     if x[0] > d.get(x[1],-1): 
      d[x[1]] = x[0] 

    y = [] 
    for k in d: 
     y.append((d[k],k)) 
    y.sort() 
    return y 

Answer 4

>>> from collections import defaultdict 
>>> d = defaultdict(tuple) 
>>> x = [(1,2), (3,4), (7,4), (5,4)] 
>>> for a, b in x: 
...  d[b] = max(d[b], (a, b)) 
... 
>>> d.values() 
[(1, 2), (7, 4) 

Answer 5

Si se puede hacer la suposición de que tuplas con idéntica segundos elementos aparecen en orden contiguo en la lista original x, puede aprovechar itertools.groupby:

import itertools 
import operator 

def max_first_elem(x): 
    groups = itertools.groupby(x, operator.itemgetter(1)) 
    y = [max(g[1]) for g in groups] 
    return y 

Tenga en cuenta que esto garantizará la preservación del orden de los grupos (por el segundo elemento de tupla), si esa es una restricción deseada para la salida.