在python中对0和1进行分组

Question

I have an array of zeros and ones like [0,0,0,1,1,1,0,0,0,1,1] . How to write a program to save neighboring 0's and 1's in different arrays.

Eg [0,0,0,1,1,1,0,0,0,1,1] giving [0,0,0],[1,1,1],[0,0,0],[1,1] .

Answer 1

You can group them with itertools.groupby , like this

>>> data = [0, 0, 0, 1, 1, 1, 0, 0, 0, 1, 1]
>>> from itertools import groupby
>>> [list(group) for item, group in groupby(data)]
[[0, 0, 0], [1, 1, 1], [0, 0, 0], [1, 1]]

The result of groupby , will be a tuple of actual item and an iterator which gives the grouped items. We just convert the grouped items to a list, with list(group) .

---

As per the comments,

>>> data = [1, 2, 1, 3, 4, 5, 6, 7, 1, 2, 3]
>>> flag = [0, 0, 0, 1, 1, 1, 0, 0, 0, 1, 1]

Create a generator which will give the values multiplied,

>>> gen = (d * v for d, v in zip(data, flag))

Now, group based on the result of calling bool on each of the numbers. So, if the bool is called on 0 , it will give False , otherwise True .

>>> [list(g) for _, g in groupby(gen, key=bool)]
[[0, 0, 0], [3, 4, 5], [0, 0, 0], [2, 3]]