Menge nach Attributen gruppieren

[sprudel]

Hallo,
mal angenommen ich habe folgende Menge:

m[0].hash = "abc"
m[1].hash = "def"
m[2].hash = "abc"
m[3].hash = "abc"
m[4].hash = "def"
m[5].hash = "xyz"

Ich möchte diese nun gruppieren, so dass die jeweiligen Objekte (!) in getrennte Listen kommen. Es geht allein um den Wert im Attribut .hash . Der Rest ist unentscheidend.

Hat da jemand ne Idee, wie man das am cleversten löst?

Liebe Grüße
Chris

[anogayales]

Pack sie doch einfach ein ein Dictionary, das Listen enthält.

Code: Alles auswählen

grp = {}
for element in m:
	try:
		grp[element.hash].append(element)
	except KeyError:
		grp[element.hash] = []
		grp[element.hash].append(element)

Grüße,
anogayales

[cofi]

Das geht noch kuerzer:

Code: Alles auswählen

grp = {}
for element in m:
    grp.setdefault(element.hash, []).append(element)

[lunar]

Und es geht auch schöner, entweder mit "defaultdict" ...

Code: Alles auswählen

from collections import defaultdict
grp = defaultdict(list)
for element in m:
    grp[element.hash].append(element)

... oder mit "itertools.groupby":

Code: Alles auswählen

from itertools import groupby
from operator import attrgetter
grp = {}
for key, items in itertools.groupby(m, attrgetter('hash')):
    grp[key] = list(items)

[BlackJack]

@lunar: Für `groupby()` müsstest Du noch sortieren.

[snafu]

`defaultdict` dürfte sich hier wohl am noch am besten für die Aufgabe eignen.

[lunar]

@BlackJack: Stimmt, daran hatte ich nicht gedacht. Mein Fehler, Verzeihung.