将列表分成不均匀的组？

Question

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

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

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

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

现在我要做的是通过 second_list 中的间距将mylist 分成不均匀的组。所以，我希望我的第一组是mylist 的前两个元素，第二组是mylist 的下四个元素，第三组是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

Padraic 拥有最好的 IMO 解决方案，但我会用这个不需要导入的 hacky 单线来补充：

>>> 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]]

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

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])]

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

>>> 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 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

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 6

此解决方案会跟踪您编写的项目数量。如果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 中找到的长度的子列表。