Python: warunkowo usuwać elementy z listy

Question

Załóżmy, że mam listę krotek:

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

Spośród wszystkich krotek, które dzielą drugi element, chcę zachować krotki z największą pierwszego elementu:

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

Jaki jest najlepszy sposób na osiągnięcie tego w Pythonie?

---

Dzięki za odpowiedzi.

• Krotki mogą być zamiast tego listami dwuelementowymi, jeśli to robi różnicę.
• Wszystkie elementy są nieujemnymi liczbami całkowitymi.
• Podoba mi się bieżące odpowiedzi. Naprawdę powinienem dowiedzieć się więcej o tym, co ma do zaoferowania collections!

Answer 1

podobne do odpowiedzi Aarona

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

Zauważ, że kolejność nie jest zachowana przy użyciu tej metody. Aby zachować zamówienie, użyj tego zamiast tego w inny sposób. Używa więcej pamięci, ale ma mniej połączeń na max, więc może to być szybciej

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

Znowu proste modyfikacje zachowuje kolejność

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

Answer 2

użycie 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

Moje własne próby, lekko inspirowany aaronsterling:

(oh yeah, wszystkie elementy są nieujemne)

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

Jeśli można zrobić założenie, że krotki z identyczne drugie elementy pojawiają się w ciągłej kolejności na pierwotnej liście x można wykorzystać 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 

pamiętać, że to gwarantuje zachowanie kolejności grup (przez drugi element krotki), jeśli jest to pożądane ograniczenie dla wyjścia.