如何在Python 3中分块列表？

Question

我找到了一段可以在Python 2中使用的代码。

from itertools import izip_longest
def grouper(n, iterable, padvalue=None):
  "grouper(3, 'abcdefg', 'x') --> ('a','b','c'), ('d','e','f'), ('g','x','x')"
  return izip_longest(*[iter(iterable)]*n, fillvalue=padvalue)

但是，这段代码在Python 3中不管用。我遇到了以下错误：

ImportError: cannot import name izip_longest

有人能帮忙吗？

我想把我的列表 [1,2,3,4,5,6,7,8,9] 转换成 [[1,2,3],[4,5,6],[7,8,9]] 这样的格式。

编辑

现在兼容Python 3

下面的代码是根据选中的答案修改的。只需把名字从 izip_longest 改成 zip_longest 就可以了。

from itertools import zip_longest
def grouper(n, iterable, padvalue=None):
  "grouper(3, 'abcdefg', 'x') --> ('a','b','c'), ('d','e','f'), ('g','x','x')"
  return zip_longest(*[iter(iterable)]*n, fillvalue=padvalue)

Answer 1

根据文档的说明：

>>> s = [1,2,3,4,5,6,7,8,9]
>>> n = 3
>>> list(zip(*[iter(s)]*n))
[(1, 2, 3), (4, 5, 6), (7, 8, 9)]

Answer 2

经过上面的讨论，这里有一个我认为能提供更安全、更可预测结果的 Python 3 解决方案。

def chunker(iter, size):
    chunks = [];
    if size < 1:
        raise ValueError('Chunk size must be greater than 0.')
    for i in range(0, len(iter), size):
        chunks.append(iter[i:(i+size)])
    return chunks

example = [1,2,3,4,5,6,7,8,9]
print(' 1: ' + str(chunker(example, 1)))
print(' 3: ' + str(chunker(example, 3)))
print(' 4: ' + str(chunker(example, 4)))
print(' 8: ' + str(chunker(example, 8)))
print(' 9: ' + str(chunker(example, 9)))
print('10: ' + str(chunker(example, 10)))

结果是：

$ python3 iter_chunk.py 
 1: [[1], [2], [3], [4], [5], [6], [7], [8], [9]]
 3: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
 4: [[1, 2, 3, 4], [5, 6, 7, 8], [9]]
 8: [[1, 2, 3, 4, 5, 6, 7, 8], [9]]
 9: [[1, 2, 3, 4, 5, 6, 7, 8, 9]]
10: [[1, 2, 3, 4, 5, 6, 7, 8, 9]]

Answer 3

在Python 3的itertools模块里，有一个叫zip_longest的函数。它的功能和Python 2里的izip_longest是差不多的。

为什么名字会变呢？你可能还会发现，Python 3里没有了itertools.izip这个函数。这是因为在Python 3中，内置的zip函数现在返回的是一个迭代器，而在Python 2中返回的是一个列表。既然izip这个函数不再需要了，那么把_longest这个变体改个名字也是为了保持一致性。