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

Question

У меня есть список массивов с каждой формой массива (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)

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

Answer 1

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

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

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