当前位置: 首页 > 面试题库 >

数组的子序列不为零

潘高岑
2023-03-14
问题内容

我正在尝试编写代码以获取此形式的列表的子序列:

l = [240,200,160,4,0,0,0,0,4,4,4,0,0,0,1,1,1,1]

基本上,我需要获取不为零的子序列,因此需要以下形式的输出:

output = [[255,200,160,4] , [4,4,4] , [1,1,1,1]]

谢谢。


问题答案:

连续的子序列?寻呼Groupby博士,博士itertools.groupby

>>> from itertools import groupby
>>> l = [240,200,160,4,0,0,0,0,4,4,4,0,0,0,1,1,1,1]
>>> [list(g) for k,g in groupby(l, lambda x: x != 0) if k]
[[240, 200, 160, 4], [4, 4, 4], [1, 1, 1, 1]]

甚至我们利用事实bool(0)Falsebool(any other integer)True

>>> [list(g) for k,g in groupby(l, bool) if k]
[[240, 200, 160, 4], [4, 4, 4], [1, 1, 1, 1]]


 类似资料: