how to extract all sublists from nested list?

Question

i have a nested list. For example:

['a', ['b', 'c', ['e', 'd']]]

I want to get a list which contains that list and all sublists separately as elements. So the expected results is:

[['a', ['b', 'c', ['e', 'd']]], ['b', 'c', ['e', 'd']], ['e', 'd']]

i wrote this function:

def extract(lst):
    result = []
    result.append(lst)
    for i in lst:
        if isinstance(i, list):
            result.append(i)
            extractt(i)
    return result

But the result is not what expected. how could i fix it?

Ajax1234 · Accepted Answer · 2019-11-03 00:55:29Z

3

You can use recursion with a generator:

def get_lists(d):
  if isinstance(d, list):
     yield d
     for i in d:
        yield from get_lists(i)

print(list(get_lists(['a', ['b', 'c', ['e', 'd']]])))

Output:

[['a', ['b', 'c', ['e', 'd']]], ['b', 'c', ['e', 'd']], ['e', 'd']]

answered Nov 3, 2019 at 0:55

Ajax1234

71.7k9 gold badges67 silver badges110 bronze badges

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

Comments

Qq0 · Accepted Answer · 2019-11-03 01:05:40Z

0

I believe your code discards the result of the recursive call to extract. You could amend it like so:

def extract(lst):
    result = [lst]
    for i in lst:
        if isinstance(i, list):
            result += extract(i)
    return result

answered Nov 3, 2019 at 1:05

Qq0

9477 silver badges12 bronze badges

Collectives™ on Stack Overflow

how to extract all sublists from nested list?

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related