Question

假设有一个列表X和另一个列表num_items，它们指定了子列表中应该包含的项目数，我可以手动拆分列表：

>>> x = list(range(10))
>>> x
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

>>> num_items = [4, 4, 2]

>>> slice1 = x[:num_items[0]]
>>> slice2 = x[len(slice1):len(slice1)+num_items[1]]
>>> slice3 = x[len(slice1)+len(slice2):]
>>> slice1, slice2, slice3
([0, 1, 2, 3], [4, 5, 6, 7], [8, 9])

有两种情况，最后几个切片可能会出现问题，例如：但是我可以用空列表解决这个问题，因为我手动编写了3个切片：

>>> num_items = [9, 1, 1]
>>> slice1 = x[:num_items[0]]
>>> slice2 = x[len(slice1):len(slice1)+num_items[1]]
>>> slice3 = x[len(slice1)+len(slice2):]
>>> slice1, slice2, slice3
([0, 1, 2, 3, 4, 5, 6, 7, 8], [9], [])

如果有4个切片怎么办，例如：

>>> num_items = [9, 1, 1, 2]
>>> slice1 = x[:num_items[0]]
>>> slice2 = x[len(slice1):len(slice1)+num_items[1]]
>>> slice3 = x[len(slice1)+len(slice2):len(slice1)+len(slice2)+num_items[2]]
>>> slice4 = x[len(slice1)+len(slice2)+len(slice3): len(slice)+len(slice2)+len(slice3)+num_items[3]]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: object of type 'type' has no len()

所需的输出是将空列表添加到第4个切片，即：

>>> slice1, slice2, slice3, slice4
([0, 1, 2, 3, 4, 5, 6, 7, 8], [9], [], [])

如果num_items要求的项目少于X的长度，则只需返回num_items的总和，即

>>> num_items = [4, 4]
>>> slice1, slice2
([0, 1, 2, 3], [4, 5, 6, 7])

主要问题是有没有办法在不手动编码分割的情况下拆分切片？（覆盖num_items请求更多项目而不是X的情况的答案，在这种情况下，应返回空的子列表）

请记住，X的长度可能相当大（即> 10,000,000,000），但num_items的长度范围为1到100 =）

Answer 1

粗暴但简单的方法。

>>> x = list(range(10))
>>> num_items = [2,3,4,1]
>>> cur_sum = 0
>>> slices = []
>>> for i in num_items:
...     slices.append(x[cur_sum:cur_sum+i])
...     cur_sum += i
... 
>>> slices
[[0, 1], [2, 3, 4], [5, 6, 7, 8], [9]]

Answer 2

这里有不同的方法：

[[x.pop(0) for _ in x[:s]] for s in num_items]

<强>示例：

>>> x = range(10)
>>> n = [9, 1, 1]
>>> [[x.pop(0) for y in x[:s]] for s in n]
[[0, 1, 2, 3, 4, 5, 6, 7, 8], [9], []]

>>> x = range(10)
>>> n = [2, 2, 2, 2, 2, 2]
>>> [[x.pop(0) for y in x[:s]] for s in n]
[[0, 1], [2, 3], [4, 5], [6, 7], [8, 9], []]

>>> x = range(10)
>>> n = [3, 4, 5, 2]
>>> [[x.pop(0) for y in x[:s]] for s in n]
[[0, 1, 2], [3, 4, 5, 6], [7, 8, 9], []] # Notice here how slice 5 only returns 3 numbers because there are only 3 numbers left in x

Answer 3

你可以这样做：

>>> x = list(range(10))
>>> num_items = [9, 1, 1]
>>> s = 0
>>> for i in num_items:
...     print x[s:s + i]
...     s += i
...

打印：

[0, 1, 2, 3, 4, 5, 6, 7, 8]
[9]
[]

将列表拆分为另一个列表

3 个答案: