将列表分成不均匀的组？

Question

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

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

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

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

现在我要做的是将mylist按second_list中的间距分成不均匀的组。所以，我希望我的第一组是 [=12= 的前 2 个元素，第二组是 [=12= 的后 4 个元素]，第三组是 [=12= 的后 5 个元素]]，第四组是`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

此解决方案会跟踪您编写的项目数。如果 second_list 中的数字总和比 mylist

长，它将崩溃

total = 0
listChunks = []
for j in range(len(second_list)):
    chunk_mylist = mylist[total:total+second_list[j]]
    listChunks.append(chunk_mylist)
    total += second_list[j]

在运行之后，listChunks 是一个包含子列表的列表，其长度在 second_list 中找到。

Answer 2

将列表推导与切片和 sum() 函数一起使用（python 的所有 基本和内置 工具）：

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

[mylist[sum(seclist[:i]):sum(seclist[:i+1])] for i in range(len(seclist))]

#output:
[[1, 2], [3, 4, 5, 6], [7, 8, 9, 10]]

如果 seclist 很长并且您希望更有效地使用 numpy.cumsum() 首先：

import numpy as np
cumlist = np.hstack((0, np.cumsum(seclist)))
[mylist[cumlist[i]:cumlist[i+1]] for i in range(len(cumlist)-1)]

并得到相同的结果

Answer 3

subgroups = []
start=0
for i in second_list:
  subgroups.append(mylist[start:start + i])
  start = i + start

最后 subgroups 将包含所需的列表

示例运行：

>>> mylist = [1,2,3,4,5,6,7,8,9,10,11,12]
>>> second_list = [2,4,5,9]
>>> subgroups = []
>>> start=0
>>> for i in second_list:
...    subgroups.append(mylist[start:start + i])
...    start = i + start
...
>>> subgroups
[[1, 2], [3, 4, 5, 6], [7, 8, 9, 10, 11], [12]]

Answer 4

您可以创建一个迭代器并且 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], [], []]

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

Answer 5

一种 numpythonic 方法：

>>> lst = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
>>> sec = [2, 4, 5]
>>> np.split(lst, np.cumsum(sec))
[array([0, 1]), array([2, 3, 4, 5]), array([ 6,  7,  8,  9, 10]), array([11])]

这里是 Python3.X 使用 itertool.accumulate() 的方法：

>>> lst = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
>>> sec = [2,4,6]
>>> from itertools import accumulate
>>> sec = list(accumulate(sec_lst))
>>> sec = [0] + sec + [None] if sec[0] != 0 else sec + [None]
>>> 
>>> [lst[i:j] for i, j in zip(sec, sec[1:])]
[[0, 1], [2, 3, 4, 5], [6, 7, 8, 9, 10], [11]]

Answer 6

Padraic 有最好的解决方案 IMO，但我会用这个不需要进口的 hacky one liner 插话：

>>> L = [1,2,3,4,5,6,7,8,9,10] # source list - could be any iterable
>>> S = [2,3,4] # group sizes - could be any iterable

>>> [ [ [ next(i) for k in range(j) ] for j in iter(S) ] for i in [iter(L)] ][0]
[[1, 2], [3, 4, 5], [6, 7, 8, 9]]

将列表分成不均匀的组？

Splitting a list into uneven groups?

python

list

python-2.7

split

sublist