Как изменить форму массивов numpy внутри списка и добавить / расширить результаты? - PullRequest
0 голосов
/ 07 августа

У меня есть список массивов с каждой формой массива (2, 2). Я хочу «разделить» их по строкам, чтобы каждая строка добавлялась в новый список (1, 2).

Пример:

X = np.random.rand(2, 2)
Y = np.random.rand(2, 2)
li = [X, Y]
li
[array([[0.06959048, 0.14011019],
        [0.8171005 , 0.31335564]]),
 array([[0.94952885, 0.15987638],
        [0.8033848 , 0.56760287]])]

Мне нужен этот результат:

new_li
[array([0.06959048, 0.14011019]),
 array([0.8171005 , 0.31335564]),
 array([0.94952885, 0.15987638]),
 array([0.8033848 , 0.56760287])]

Я знаю, что extension для каждой переменной даст желаемый результат, например:

li.extend(X)
li.extend(Y)

Но у меня есть уже существующий список.

1 Ответ

1 голос
/ 07 августа

Используйте следующее:

new_li = []
new_li.extend(j for i in li for j in i)

Это разделит каждую строку, а затем расширит все.

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