Separate a List in 2 or more lists

Question

If I have the following list of lists for example:

 [['A', 'B', 'C'], ['A', 'D', 'E', 'B', 'C']]

How could I get a List with lists of only 3 elems each (in case they are greater than 3 elems), if they have not more than 3 elems we don't need to do nothing, we just need to separate the elems with more than 3 like the following:

[['A', 'B', 'C'], ['A', 'D', 'E'], ['D', 'E', 'B'], ['E', 'B', 'C']]

Could you help me with this ? I've been trying for a long time without success, kinda new to Python.

Edit: Well, I resolved this in this way:

def separate_in_three(lista):
    paths = []
    for path in lista:
        if len(path) <= 3:
            paths.append(path)
        else:
            for node in range(len(path)-1):
                paths.append(path[:3])
                path.pop(0);
                if(len(path) == 3):
                    paths.append(path)
                    break
    return paths

Seems to resolve my problem, I could use the list in comprehension, were it would be much more efficient than the way I did ? Thanks for the help btw !

What have you searched for, and what did you find? What have you tried, and how did it fail? — tripleee, Mar 20 '21 at 16:11
try taking a look at this answer: https://stackoverflow.com/a/9671301/15386705 — sortBot, Mar 20 '21 at 16:13
I think answer [this](https://stackoverflow.com/a/31054676/13745968) can help you. — Higo Felipe Silva Pires, Mar 20 '21 at 16:15

score 0 · Answer 1 · answered Mar 20 '21 at 16:11

0

you can use list comprehension like below.

l =  [['A', 'B', 'C'], ['A', 'D', 'E', 'B', 'C','Z']]
[l[0]] + [l[1][i: i+len(l[0])] for i in range(1 + len(l[1]) - len(l[0]))]

answered Mar 20 '21 at 16:11

Paras Gupta

174
4

Thanks! It works for that example but it's not general for all lists ofc, but easily adapted for that, much appreciated ! – Ezequiel Barreira Mar 20 '21 at 21:52

Separate a List in 2 or more lists

1 Answers1