python - 根据 Python 中的一组索引将列表拆分为多个部分

Question

根据任意数量的索引将列表拆分成多个部分的最佳方法是什么？例如。给出下面的代码

indexes = [5, 12, 17]
list = range(20)

返回这样的东西

part1 = list[:5]
part2 = list[5:12]
part3 = list[12:17]
part4 = list[17:]

如果没有索引，它应该返回整个列表。

Answer 1

这是我能想到的最简单也是最pythonic的解决方案:

def partition(alist, indices):
    return [alist[i:j] for i, j in zip([0]+indices, indices+[None])]

如果输入非常大，那么迭代器解决方案应该更方便:

from itertools import izip, chain
def partition(alist, indices):
    pairs = izip(chain([0], indices), chain(indices, [None]))
    return (alist[i:j] for i, j in pairs)

当然，还有非常非常懒惰的家伙解决方案(如果您不介意获取数组而不是列表，但无论如何您总是可以将它们还原为列表):

import numpy
partition = numpy.split