Nested list comprehension with if else to avoid iterating over integers

Asked Aug 13 '16 at 19:36

Active Sep 07 '22 at 07:37

Viewed 21 times

lets say i have a list which contains lists and integers. I want to write a comprehension where i get a single list of all elements as individual integers. Eg lst = [3,4,5,6,4,[2,3,4,9],[2,3,4]] which i want to convert into a single list [3,4,5,6,4,2,3,4,9,2,3,4]. Best I could manage so far is:

lst = [3,4,5,6,4,[2,3,4,9],[2,3,4]]
a = [i for i in lst if type(i)==int]
b = [j for i in lst if type(i)==list for j in i ]
x = a + b

is there a way to do this in a single list comprehension? Furthermore is there a convenient way to get a single list regardless of the input list structure? eg. inputList = [1,[2,3],[2,[3,4]]] outputList = [1,2,3,2,3,4] Thanks in advance for response.

asked Aug 13 '16 at 19:36

commonsensei

You need recursion or a stack to flatten arbitrary levels of nesting – Padraic Cunningham Aug 13 '16 at 19:39
duplicate of http://stackoverflow.com/questions/952914/making-a-flat-list-out-of-list-of-lists-in-python – Jean-François Fabre Aug 13 '16 at 19:40
http://stackoverflow.com/q/2158395/1005215 – Nehal J Wani Aug 13 '16 at 19:40
would you be able to shed some light on if and how the specific example can be solved with a single list comprehension? – commonsensei Aug 13 '16 at 19:43

Nested list comprehension with if else to avoid iterating over integers

0 Answers0