Das deutsche Python-Forum

hallo!

wieder mal eine kleine frage von mir:

ich hab folgende liste:

liste = [['elem 1','elem 2','elem 3'],
['elem 1','elem 2','elem 3','elem 4'],
['elem 1','elem 2','elem 3','elem 4','elem 5' ],
['elem a 1','elem a 2','elem a 3'],
['elem a 1','elem a 2','elem a 3','elem a 4']
]

Meine Liste ist länger, aber diese kurze reicht um die Frage zu stellen.
Wie kann ich die Liste zuammenfassen, dass nur die jeweils längsten Einträge von elem und elem a überbleiben:

Code: Alles auswählen


new_list =[ ['elem 1','elem 2','elem 3','elem 4','elem 5' ], 
['elem a 1','elem a 2','elem a 3','elem a 4']]

Danke,

Johannes

Code: Alles auswählen

from itertools import groupby
from pprint import pprint

data = [['elem 1','elem 2','elem 3'],
        ['elem 1','elem 2','elem 3','elem 4'],
        ['elem 1','elem 2','elem 3','elem 4','elem 5' ],
        ['elem a 1','elem a 2','elem a 3'],
        ['elem a 1','elem a 2','elem a 3','elem a 4']]

def keyfunc(item):
    return item[0].rsplit(' ', 1)[0]    # Get the element name.

data.sort(key=keyfunc)  # If not already sorted.
result = list()
for key, items in groupby(data, keyfunc):
    result.append(max((len(item), item) for item in items)[1])

pprint(result)

Das übliche `groupby()` und dann werden Tupel mit Listenlänge und zugehöriger Liste gebildet, daraus dann das Maximum ermittelt, also die längste bestimmt, und dann aus dem "grössten" Tupel die Liste als Ergebnis genommen.

Das deutsche Python-Forum

frage um 2D liste zusammenzufassen

frage um 2D liste zusammenzufassen