將列表從“ 0”拆分為兩個列表

Question

我有一個清單：

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

需要將其拆分為0個隨機列表（忽略0）：

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

我將如何去做呢？ 原始列表將具有隨機長度且具有隨機數。

Answer 1

一種選擇是使用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]]