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