Flatten 3d list to 2d in python

Question

So I have a list of lists in python that is something like this:

[[[0, 1, 0, 1, 0]]
[[1, 1, 1, 1, 1]]
[[1, 0, 0, 1, 1]]
[[0, 1, 0, 0, 0]]]

I want to flatten this list and end up with this:

[[0, 1, 0, 1, 0]
[1, 1, 1, 1, 1]
[1, 0, 0, 1, 1]
[0, 1, 0, 0, 0]]

Is there a straightforward way to do this in python?

Could you share a more copy/paste friendly version of the lists? — yatu
– yatu, Commented Jul 12, 2019 at 11:00

zipa · Accepted Answer · 2019-07-12 11:09:07Z

2

Using numpy.squeeze you can do what you want:

import numpy as np

a = np.array([[[0, 1, 0, 1, 0]],
              [[1, 1, 1, 1, 1]],
              [[1, 0, 0, 1, 1]],
              [[0, 1, 0, 0, 0]]])

a.squeeze()

[[0 1 0 1 0]
 [1 1 1 1 1]
 [1 0 0 1 1]
 [0 1 0 0 0]]

edited Jul 12, 2019 at 11:09

answered Jul 12, 2019 at 11:07

zipa

28k6 gold badges45 silver badges62 bronze badges

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

1 Comment

zipa Over a year ago

@yatu I know, there is already answer for list, so I converted list to np.array

frankegoesdown · Accepted Answer · 2019-07-12 11:07:37Z

2

a = [[[0, 1, 0, 1, 0]], 
[[1, 1, 1, 1, 1]], 
[[1, 0, 0, 1, 1]], 
[[0, 1, 0, 0, 0]]]    

[i[0] for i in a]

output

[[0, 1, 0, 1, 0], [1, 1, 1, 1, 1], [1, 0, 0, 1, 1], [0, 1, 0, 0, 0]]

edited Jul 12, 2019 at 11:07

answered Jul 12, 2019 at 11:01

frankegoesdown

1,9241 gold badge22 silver badges40 bronze badges

Collectives™ on Stack Overflow

Flatten 3d list to 2d in python

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related