defaultdict(list) sortieren

[mit]

Hallo,
Ich habe ein defauldict(list) wobei die Liste beinhaltet dicts.

Code: Alles auswählen

from operator import itemgetter
from collections import defaultdict
import pprint

a = defaultdict(list)

a[1] = [{ 'a':3, 'b':1, 'c':3},{'a':2, 'b':1, 'c':3}, { 'a':0, 'b':1, 'c':3}]
a[2] = [{ 'a':6, 'b':1, 'c':3},{'a':4, 'b':1, 'c':3}, { 'a':5, 'b':1, 'c':3}]

for key in a:
    a[key] = sorted(a[key], key=itemgetter('a')) 
    
pprint.pprint(a)

Ich war in der lange die Liste zu sortieren aber ich bin mir nicht sicher ob es die beste Loesung ist?

[pillmuncher]

Es ist schon fast optimal. So ist es noch besser:

Code: Alles auswählen

from operator import itemgetter
from collections import defaultdict
import pprint

a = defaultdict(list)

a[1] = [{ 'a':3, 'b':1, 'c':3},{'a':2, 'b':1, 'c':3}, { 'a':0, 'b':1, 'c':3}]
a[2] = [{ 'a':6, 'b':1, 'c':3},{'a':4, 'b':1, 'c':3}, { 'a':5, 'b':1, 'c':3}]

a_key = itemgetter('a')

for each in a.values():
    each.sort(key=a_key)

pprint.pprint(a)

[mit]

Danke, es funktioniert.

Wie könnte man nur den höchsten Wert speichern?

[EyDu]

Du meinst analog zur sort-Funktion? Mittels max.

[mit]

Habe ich versucht, aber ich bekomme "each.max(key=a_key) AttributeError: 'list' object has no attribute 'max'".

Code: Alles auswählen

    from operator import itemgetter
    from collections import defaultdict
    import pprint
     
    a = defaultdict(list)
     
    a[1] = [{ 'a':3, 'b':1, 'c':3},{'a':2, 'b':1, 'c':3}, { 'a':0, 'b':1, 'c':3}]
    a[2] = [{ 'a':6, 'b':1, 'c':3},{'a':4, 'b':1, 'c':3}, { 'a':5, 'b':1, 'c':3}]
     
    a_key = itemgetter('a')
     
    for each in a.values():
        each.max(key=a_key)
     
    pprint.pprint(a)

[pillmuncher]

@mit:

Code: Alles auswählen

>>> max_items = [max(each, key=a_key) for each in a.values()]
>>>
>>> pprint.pprint(max_items)
[{'a': 3, 'b': 1, 'c': 3}, {'a': 6, 'b': 1, 'c': 3}]

[mit]

Danke, es funktioniert fast, aber wie bekommt folgendes hin?

Code: Alles auswählen

{1: [{'a': 3, 'c': 3, 'b': 1}], 2: [{'a': 6, 'c': 3, 'b': 1}]}
or
defaultdict(<type 'list'>, {1: [{'a': 3, 'c': 3, 'b': 1}], 2: [{'a': 6, 'c': 3, 'b': 1}]})

[EyDu]

Was soll denn da passieren? Willst du die Einträge durchnummerieren? Ich hoffe doch nicht, denn dann solltest du Listen oder Tupel verwenden.

[pillmuncher]

hm?

[mit]

Ich kam ab mit dieser loesung, aber ich weiss nicht ob es die schnellste ist:

Code: Alles auswählen

from operator import itemgetter
from collections import defaultdict
import pprint
 
a = defaultdict(list)
 
a[1] = [{ 'a':3, 'b':1, 'c':3},{'a':2, 'b':1, 'c':3}, { 'a':0, 'b':1, 'c':3}]
a[2] = [{ 'a':6, 'b':1, 'c':3},{'a':4, 'b':1, 'c':3}, { 'a':5, 'b':1, 'c':3}]
 
a_key = itemgetter('a')
 
for each in a.values():
    each.sort(key=a_key)
 
pprint.pprint(a)

# verkleinern
b = defaultdict(dict)
for k, v in a.items():
    b[k] = v[0]

pprint.pprint(b)

[Hyperion]

Bezüglich der Zeitkomplexität handelst Du Dir durch das Sortieren O(n * log n) ein. Das ließe sich optimieren, indem Du einfach die ``min``-Funktion für jede Value-Liste in ``a`` anwendest. Die hat nur O(n), ist also bei vielen Elementen deutlich schneller:

Code: Alles auswählen

# mal für ein value-Element:
min(a[1], key=itemgetter("a"))
>  {'a': 0, 'b': 1, 'c': 3}

Edit: Aufbauend auf dem bisherigen Code komme ich darauf:

Code: Alles auswählen

b = {k: min(v, key=a_key) for k, v in a.items()}

[BlackJack]

@mit: Das schnellste wofür? Falls Du nur `b` als Endergebnis haben möchtest und die Listen nirgends wirklich sortiert benötigst, dann wäre `max()` (für die ursprünglich gezeigten gewünschten Ergebnisdaten) oder `min()` (für das Ergebnis des letzten Codebeispiels) schneller, weil dann nur linear durch die Listen gegangen werden muss statt sie zu sortieren.

[mit]

Danke es funktioniert;

Code: Alles auswählen

from operator import itemgetter
from collections import defaultdict
import pprint
 
a = defaultdict(list)
 
a[1] = [{ 'a':3, 'b':1, 'c':3},{'a':2, 'b':1, 'c':3}, { 'a':0, 'b':1, 'c':3}]
a[2] = [{ 'a':6, 'b':1, 'c':3},{'a':4, 'b':1, 'c':3}, { 'a':5, 'b':1, 'c':3}]
 
# verkleinern
a_key = itemgetter('a')
b = defaultdict(dict)
b = {k: min(v, key=a_key) for k, v in a.items()}

[BlackJack]

@mit: Zeile 12 ist sinnfrei.