Python: How do I slice a list based on variable type

Question

I am trying to slice a list into a list of lists, whereby the original list is sliced at the point where a string type object is detected e.g.

array = ['Hello',1,2,3,'Goodbye',4,6,'Bye',7,8,9,5]

to

new_array = [['Hello',1,2,3],['Goodbye',4,6],['Bye',7,8,9,5]]

How could this be achieved?

user2390182 · Accepted Answer · 2020-05-27 10:29:46Z

0

You can use a generator function for this, e.g.:

def gen(arr):
    r = []
    for el in arr:
        if r and isinstance(el, str):
            yield r
            r = []
        r.append(el)
    if r:
        yield r

>>> list(gen(array))
[['Hello', 1, 2, 3], ['Goodbye', 4, 6], ['Bye', 7, 8, 9, 5]]

answered May 27, 2020 at 10:29

user2390182

73.7k6 gold badges71 silver badges95 bronze badges

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

2 Comments

user9697260 Over a year ago

Hi thanks so much! That's really interesting. I've never come across generator functions before. Do you have a good website that you'd recommend that discusses them?

user2390182 Over a year ago

The official docs should get you a long part of the way: docs.python.org/3/howto/functional.html#generators

Collectives™ on Stack Overflow

Python: How do I slice a list based on variable type

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related