Python列表划分/拆分[重复]

Question

可能重复：
How do you split a list into evenly sized chunks in Python?

你好，

我正在尝试找到一种更简单的方法来执行以下操作：

def list_split(list, size):
  result = [[]]
  while len(list) > 0:
    if len(result[-1]) >= size: result.append([])
    result[-1].append(list.pop(0))
  return result

示例用法：

>>> list_split([0, 1, 2, 3, 4, 5, 6], 2)
[[0, 1], [2, 3], [4, 5], [6]]
>>> list_split([0, 1, 2, 3, 4, 5, 6], 3)
[[0, 1, 2], [3, 4, 5], [6]]

我不知道是否有内置的方法可以做到这一点，可能是切片或其他方法。

这与How to split a list into a given number of sub-lists in python的帖子相似但不一样

谢谢

编辑： 正如 Anurag Uniyal 所评论的，这是 How do you split a list into evenly sized chunks? 的副本，应该关闭，我不能这样做。

Answer 1

您可以使用切片来获取列表的子集。

例子：

>>> L = [0, 1, 2, 3, 4, 5, 6]
>>> n = 3
>>> [L[i:i+n] for i in range(0, len(L), n)]
[[0, 1, 2], [3, 4, 5], [6]]
>>>

Answer 2

您在 itertools 配方 (grouper) 中有一个简单的功能解决方案：

http://docs.python.org/library/itertools.html#recipes

虽然这个函数添加了填充，但您可以轻松地编写一个非填充的实现，利用（通常被忽略的）iter 内置以这种方式使用：iter(callable, sentinel) -> iterator

def grouper(n, it):
  "grouper(3, 'ABCDEFG') --> ABC DEF G"
  return iter(lambda: list(itertools.islice(it, n)), [])

list(grouper(3, iter(mylist)))

这些解决方案更通用，因为它们都适用于序列和可迭代对象（即使它们是无限的）。

Answer 3

def list_split(L, size):
    return [L[i*size:(i+1)*size] for i in range(1+((len(L)-1)//size))]

如果您更喜欢生成器而不是列表，可以将括号替换为括号，如下所示：

def list_split(L, size):
    return (L[i*size:(i+1)*size] for i in range(1+((len(L)-1)//size)))

Answer 4

from itertools import izip_longest

def list_split(L, size):
    return [[j for j in i if j is not None] for i in izip_longest(*[iter(L)]*size)]

---

>>> list_split([0, 1, 2, 3, 4, 5, 6], 2)
[[0, 1], [2, 3], [4, 5], [6]]
>>> list_split([0, 1, 2, 3, 4, 5, 6], 3)
[[0, 1, 2], [3, 4, 5], [6]]
>>> list_split([0, 1, 2, 3, 4, 5, 6], 4)
[[0, 1, 2, 3], [4, 5, 6]]
>>> list_split([0, 1, 2, 3, 4, 5, 6], 5)
[[0, 1, 2, 3, 4], [5, 6]]