Python：从列表中获取多个列表

Question

可能的重复：
如何将列表均匀分割Python 中的块大小？

嗨，

我想将一个列表拆分为多个长度为 x 元素的列表，例如：

a = (1, 2, 3, 4, 5)

并获取：

乙 = ( (1,2), (3,4), (5,) ）

如果长度设置为 2 或：

乙 = ( (1,2,3), (4,5) ）

如果长度等于 3 ...

有没有好的方法来写这个？否则我认为最好的方法是使用迭代器来编写它......

Answer 1

我就是这样做的。迭代，但在列表理解中。注意类型是混合的；这可能是所希望的，也可能不是所希望的。

def sublist(seq, length):
    return [seq[i:i + length] for i in xrange(0, len(seq), length)]

用法：

>>> sublist((1, 2, 3, 4, 5), 1)
[(1,), (2,), (3,), (4,), (5,)]

>>> sublist([1, 2, 3, 4, 5], 2)
[[1, 2], [3, 4], [5]]

>>> sublist('12345', 3)
['123', '45']

>>> sublist([1, 2, 3, 4, 5], 73)
[[1, 2, 3, 4, 5]]

>>> sublist((1, 2, 3, 4, 5), 0)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 2, in sublist
ValueError: xrange() arg 3 must not be zero

当然，如果您愿意，您也可以轻松地使其生成 tuple - 将列表理解 [...] 替换为 tuple(... ）。您还可以将 seq[i:i + length] 替换为 tuple(seq[i:i + length]) 或 list(seq[i:i + length]) 使其返回固定类型。

Answer 2

itertools 模块文档。阅读它，学习它，喜欢它。

具体来说，从食谱部分：

import itertools

def grouper(n, iterable, fillvalue=None):
  "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
  args = [iter(iterable)] * n
  return itertools.izip_longest(fillvalue=fillvalue, *args)

给出：

>>> tuple(grouper(3, (1,2,3,4,5)))
((1, 2, 3), (4, 5, None))

这并不完全是你想要的......你不希望其中有None......所以。快速修复：

>>> tuple(tuple(n for n in t if n) for t in grouper(3, (1,2,3,4,5)))
((1, 2, 3), (4, 5))

如果您不喜欢每次都输入列表理解，我们可以将其逻辑移至函数中：

def my_grouper(n, iterable):
  "my_grouper(3, 'ABCDEFG') --> ABC DEF G"
  args = [iter(iterable)] * n
  return tuple(tuple(n for n in t if n)
       for t in itertools.izip_longest(*args))

给出：

>>> tuple(my_grouper(3, (1,2,3,4,5)))
((1, 2, 3), (4, 5))

完成。

Answer 3

来自 itertools 模块上的 python 文档：

from itertools import izip_longest
def grouper(n, iterable, fillvalue=None):
    "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
    args = [iter(iterable)] * n
    return izip_longest(fillvalue=fillvalue, *args)

示例：

>>> tuple(grouper(3, (1, 2, 3, 4, 5, 6, 7)))
((1, 2, 3), (4, 5, 6), (7, None, None))