Сгладить список массивов numpy, сохраняя при этом внутренние массивы

l = [np.array([[1,2],[3,4]]), np.array([5,6]), np.array([[7,8],[9,10],[11,12]])]

Я пытаюсь сгладить этот список массивов, но сохраняю внутренние массивы:

[[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], [11, 12]]

Я пробовал itertools.chain, np.concatenate, np.flatten, но ни один из этих вариантов не дает результата выше

np.vstack(l) ?

— 17.03.2022 14:45

17.03.2022 11:32

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

Ответы 3

Ответ принят как подходящий

Ваши массивы имеют разное количество измерений, вам необходимо убедиться, что все они двумерные:

out = np.concatenate([x[None,:] if x.ndim == 1 else x for x in l])

выход:

array([[ 1,  2],
       [ 3,  4],
       [ 5,  6],
       [ 7,  8],
       [ 9, 10],
       [11, 12]])

или с itertools.chain с выводом списка:

from itertools import chain
list(chain.from_iterable([x.tolist()] if x.ndim == 1 else x.tolist()
                         for x in l))

вывод: [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], [11, 12]]

17.03.2022 11:35

Вы не предоставили желаемый результат. Но, как я понял, вы ищете это:

x = [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], [11, 12]]
print([n for m in x for n in m])

с выходом:

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]

Пожалуйста, дайте мне знать, если это не ваш желаемый результат.

17.03.2022 11:56

В основном вам нужно сгладить каждый подмассив, объединить их и, наконец, изменить его форму.

np.concatenate(list(map(np.ravel, l))).reshape(-1, 2)

выход:

array([[ 1,  2],
       [ 3,  4],
       [ 5,  6],
       [ 7,  8],
       [ 9, 10],
       [11, 12]])

17.03.2022 13:49