объединить вложенный список по первому значению в каждом списке в python - PullRequest
0 голосов
/ 01 февраля 2020

У меня работает версия этого кода, но она кажется слишком сложной. Это упрощенная версия, на которой я застреваю. в python3.

list1 = [['a', 'b'], ['a', 'c'], ['a', 't'], ['x', 'f'], ['x', 'g'], ['d', 'z']]

z = len(list1)
for i in range(0, z):
    for j in range(i + 1, z):
        while list1[i][0] == list1[j][0]:
            list1[i] = list1[i] + [list1[j][-1]]
            if list1[j] in list1:
                list1.remove(list1[j])
                z = z - 1

Я хочу вывод.

[['a', 'b', 'c', 't'], ['x', 'f', 'g'], ['d', 'z']]

Ответы [ 2 ]

5 голосов
/ 01 февраля 2020

Моддинг у Дэррила немного:

d = {}
for head, *tail in lst:
    d.setdefault(head, [head]).extend(tail)

lst2 = list(d.values())
2 голосов
/ 01 февраля 2020

Упрощенная конструкция состоит из следующих двух шагов:

d = {}
for sublist in lst:
  d.setdefault(sublist[0], []).extend(sublist[1:])

lst2 = [[k] + v for k, v in d.items()]

print(lst2)
>>> [['a', 'b', 'c', 't'], ['x', 'f', 'g'], ['d', 'z']]

Пояснение

(1) Словарь d размещает элементы с тем же первым элементом, что и ключ словаря со значениями, соответствующими оставшимся элементам для создания:

{'a': ['b', 'c', 't'], 'x': ['f', 'g'], 'd': ['z']}

(2) Далее, для понимания списка ключ каждой записи словаря используется как первый элемент подсписка, а значение элементов - как оставшиеся элементы для получения желаемого результата

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...