Using list comprehension instead of for loop to remove nested lists without removing their elements

Asked Oct 07 '20 at 20:37

Active Oct 07 '20 at 20:37

Viewed 22 times

      lst = [1,2,3,4,[5,6], 7,8,9]
                new_lst = []
                for i in range(len(lst)):
                    if type(lst[i])==list:
                        new_lst.extend(lst[i])
                    else:
                        new_lst.append(lst[i])
                return new_lst
        and the new_lst = [1,2,3,4,5,6,7,8,9]

Is there a way to make this work for [1,2,[3,4,[5,6,[7]]],8,9,10] ? is it possible to get to work with list comprehension ?

asked Oct 07 '20 at 20:37

Ronit Chawla

1

You might consider using [`more_itertools.collapse`](https://more-itertools.readthedocs.io/en/stable/api.html#more_itertools.collapse). – 0x5453 Oct 07 '20 at 20:39
For *arbitrarily nested lists*? Probably no sane way – juanpa.arrivillaga Oct 07 '20 at 20:49
`fn = lambda lst: (fn(lst[0]) if len(lst) == 1 else ([] if len(lst) == 0 else [vv for v in lst for vv in fn(v)])) if isinstance(lst, list) else [lst]` – Andrej Kesely Oct 07 '20 at 21:12

Using list comprehension instead of for loop to remove nested lists without removing their elements

0 Answers0