Question

我发现以下代码与python2

兼容

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

修改

现在兼容Python3

以下代码根据所选答案进行调整。只需将名称从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

在Python 3的itertools中有一个名为zip_longest的函数。它应该与Python 2中的izip_longest相同。

为什么改名？您可能还会注意到{3}现在已经消失了 - 因为在Python 3中，itertools.izip内置函数现在返回一个迭代器，而在Python 2中它返回一个列表。由于不需要zip函数，因此重命名izip变量以保持一致性也是有意义的。

Answer 2

根据the doc：

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

经过以上所有讨论，下面是一个python3解决方案，我相信它可以提供更安全，更可预测的结果。

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

如何在Python 3中列出一个列表？

修改

现在兼容Python3

3 个答案: