Как полностью сгладить список в Python? [Дубликат]

Question

Как полностью сгладить список в Python? [Дубликат]

1

Сглаживание списков списков является хорошо известной проблемой, но как сгладить список списков списков наиболее пифонов? Моя конечная цель - создать массив numpy следующим образом:

[In]: mylist = [[[1,2,3], [4]], [[5,6], [7,8]]]
[Out]: array([[1, 2, 3, 4],
              [5, 6, 7, 8]])

но даже просто полностью сглаживается в длинный список (в любом случае его можно легко преобразовать в правую форму). Типичный выходной массив будет иметь размеры (10000, 10), поэтому циклическое перемещение по строкам будет очень неэффективным. Все идеи будут оценены!

(edit) Лучшее решение, о котором я могу думать:

import itertools

for i in range(len(mylist)):
    mylist[i] = list(itertools.chain.from_iterable(mylist[i]))
np.array(mylist)

Petr Jakubčík 25 авг. 2018, в 15:41

Источник

1

Просто чтобы понять проблему, как вы могли бы иметь циклическое решение для решения вашего примера?
Divakar 25 авг. 2018, в 12:57
0

На самом деле многие решения, которые выглядят заманчиво, сначала возвращают array([list([1, 2, 3]), list([4]), list([5, 6]), list([7, 8])], dtype=object)
Petr Jakubčík 25 авг. 2018, в 13:07
0

Это также один из моих любимых: stackoverflow.com/a/10632307/7919597
Joe 25 авг. 2018, в 13:20
0

list(map(list, map(itertools.chain.from_iterable, mylist))) ?
Mark Dickinson 25 авг. 2018, в 13:32

Показать ещё 2 комментария

Теги:

optimization

python

numpy

itertools

flatten

1 ответ

Ещё вопросы

Просто чтобы понять проблему, как вы могли бы иметь циклическое решение для решения вашего примера?
На самом деле многие решения, которые выглядят заманчиво, сначала возвращают array([list([1, 2, 3]), list([4]), list([5, 6]), list([7, 8])], dtype=object)
Это также один из моих любимых: stackoverflow.com/a/10632307/7919597
list(map(list, map(itertools.chain.from_iterable, mylist))) ?

Daniel Mesejo · Answer 1 · 2018-08-25T12-06-00.000Z

Возможность использования numpy.append:

import numpy as np

rows = [[[1, 2, 3], [4]], [[5, 6], [7, 8]]]
result = np.array([np.append(value[0], value[1:]) for value in rows])

print(result)

Выход

[[1 2 3 4]
 [5 6 7 8]]

Функция numpy.append выравнивает значение [1:]. Один из вариантов заключается в использовании concatenate и ravel:

rows = [[[1], [2], [3]], [[4], [5], [6]]]
result = np.array([np.concatenate(value).ravel() for value in rows])
print(result)

Выход

[[1 2 3]
 [4 5 6]]

Функция concatenate объединяет последовательность массивов, а np.ravel возвращает смежный сплющенный массив. Вы можете изменить np.ravel на np.flatten.

Хорошая идея, но она терпит неудачу, когда строки содержат более двух списков ... rows = [[[1], [2], [3]], [[4], [5], [6]]] throws Ошибка типа.
append делает много перемещений. Это не амортизируется вообще, в отличие от добавления и расширения списка. Все еще хорошая мысль.