根据值拆分Numpy数组

根据值拆分Numpy数组

您可以使用

groupby()

from中的函数

itertools

，并将

key

布尔值指定为零或非零的布尔条件。这样，所有连续的零和非零将被组合在一起。使用

if

过滤器选取非零组，

list

并将非零石斑鱼转换为列表。

from itertools import groupby[list(g) for k, g in groupby(a, lambda x: x != 0) if k]# [[3, 5], [10, 14, 15, 56], [12, 23, 45, 23, 12, 45], [1], [2, 3, 4]]