将列表从“ 0”拆分为两个列表

Question

I have a list:

matrix = [1, 2, 3, 0, 4, 5, 0, 6]

which needs to be split into a random amount of lists at 0 (ignoring 0):

matrix1 = [3, 2, 1]
matrix2 = [4, 5]
matrix3 = [6]

How would I go about doing this? The original list will be of random length and with random numbers.

Answer 1

One option is to use groupby :

from itertools import groupby

lst = [1, 2, 3, 0, 4, 5, 0, 6]
[list(g) for k, g in groupby(lst, lambda x: x != 0) if k]
# [[1, 2, 3], [4, 5], [6]]