我有一个名为值的列表,其中包含一系列数字:
values = [0, 1, 2, 3, 4, 5, ... , 351, 0, 1, 2, 3, 4, 5, 6, ... , 750, 0, 1, 2, 3, 4, 5, ... , 559]
我想创建一个新列表,其中包含从0到数字的元素列表.
喜欢 :
new_values = [[0, 1, 2, ... , 351], [0, 1, 2, ... , 750], [0, 1, 2, ... , 559]]
我做的代码是这样的:
start = 0
new_values = []
for i,val in enumerate(values):
if(val == 0):
new_values.append(values[start:i])
start = i
但是,它将返回以下内容:
new_values = [[], [0, 1, 2, ... , 750], [0, 1, 2, ... , 559]]
如何修复我的代码?这将确实有很大的帮助.
解决方法:
您可以基于0(很虚假)的存在,使用itertools.groupby对元素进行分组,并提取0之间的子列表,同时在缺失的0后面附加列表理解:
[[0]+list(g) for k, g in groupby(values, bool) if k]
例:
>>> from itertools import groupby
>>> values = [0, 1, 2, 3, 4, 5 , 351, 0, 1, 2, 3, 4, 5, 6, 750, 0, 1, 2, 3, 4, 559]
>>> [[0]+list(g) for k, g in groupby(values, bool) if k]
[[0, 1, 2, 3, 4, 5, 351], [0, 1, 2, 3, 4, 5, 6, 750], [0, 1, 2, 3, 4, 559]]
标签:python,list