使用索引列表拆分列表

Question

我试图根据来自另一个列表的索引对将列表拆分成组，给定:

>>> l = list(range(10))
>>> l
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> idx = [0, 5]

我需要将列表分解为:

>>> l[0:5]
[0, 1, 2, 3, 4]
>>> l[5:]
[5, 6, 7, 8, 9]

列表idx将至少始终为[0]，但大小可以为n；idx内的值将始终按升序排序。

目前我有:

>>> l = list(range(10))
>>> idx = [0, 5]
>>> idx.append(None)
>>> [l[idx[i]:idx[i + 1]] for i in range(len(idx) - 1)]
[[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]

有没有一种方法可以在不显式追加non和在范围上迭代的情况下实现这一点？

编辑:另一个例子。。。

给定:

>>> l = list(range(14))
>>> l
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]
>>> idx = [0, 5, 10]

所需结果:

[[[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13]]

Answer 1

您可以尝试使用itertools.zip_longth:

from itertools import zip_longest

l = list(range(14))
idx = [0, 5, 10]
print([l[pre: last] for pre, last in zip_longest(idx,idx[1:])])

结果:

[[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13]]