How to reshape numpy arrays inside a list and append/extend results?

Question

I have a list of arrays with each array shape (2, 2). I want to "divide" them by rows so that each row is added to a new list of (1, 2).

Example:

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]])]

I want this result:

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

I know that extend for each variables will do the desired result, e.g.:

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

But I have a pre-existing list.

Rahul Vishwakarma · Accepted Answer · 2020-08-07 13:47:37Z

1

Use the following:

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

This will split each row and then extend all.

answered Aug 7, 2020 at 13:47

Rahul Vishwakarma

1,4562 gold badges12 silver badges23 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to reshape numpy arrays inside a list and append/extend results?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related