python - 将列表分成不均匀的组？

Question

我知道如何将列表分成偶数组，但我无法将它分成不均匀的组。

基本上这就是我所拥有的:一些列表，我们称它为 mylist，它包含 x 个元素。

我还有另一个文件，我们称之为 second_list，它看起来像这样:

{2, 4, 5, 9, etc.}

现在我要做的是将 mylist 按 second_list 中的间距分成不均匀的组。所以，我希望我的第一组是 mylist 的前 2 个元素，第二组是 mylist 的后 4 个元素，第三组是后 5 个元素mylist 的元素，第四组是 `mylist 的下 9 个元素，依此类推。

有什么简单的方法可以做到这一点吗？如果您想将其分成偶数组，我尝试做类似的事情:

for j in range(0, len(second_list)):
    for i in range(0, len(mylist), second_list[j]):
        chunk_mylist = mylist[i:i+second_list[j]]

然而，这并没有像我想要的那样拆分它。我想以我的子列表数量为 len(second_list) 结束，并且还正确拆分，这提供的远不止于此(而且拆分不正确)。

Answer 1

你可以创建一个迭代器和itertools.islice :

mylist = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
seclist = [2,4,6]

from itertools import islice
it = iter(mylist)

sliced =[list(islice(it, 0, i)) for i in seclist]

这会给你:

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

一旦 i 个元素被消耗，它们就消失了，所以我们继续获取下一个 i 个元素。

不确定剩下的元素会发生什么，如果你想添加它们，你可以添加如下内容:

mylist = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13 ,14]
seclist = [2, 4, 6]

from itertools import islice

it = iter(mylist)

slices = [sli for sli in (list(islice(it, 0, i)) for i in seclist)]
remaining = list(it)
if remaining:
    slices.append(remaining)
print(slices)

这会给你:

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

或者相反，如果没有足够的列表，您可以使用几种方法来删除空列表，一种是内部生成器表达式:

from itertools import islice

it = iter(mylist)
slices = [sli for sli in (list(islice(it, 0, i)) for i in seclist) if sli]

或结合itertools.takewhile :

 from itertools import islice, takewhile

it = iter(mylist)
slices = list(takewhile(bool, (list(islice(it, 0, i)) for i in seclist)))

用于:

mylist = [1, 2, 3, 4, 5, 6]
seclist = [2, 4, 6,8]

会给你:

[[1, 2], [3, 4, 5, 6]]

相对于:

[[1, 2], [3, 4, 5, 6], [], []]

您使用什么完全取决于您可能的输入以及您希望如何处理各种可能性。