创建子列表

Question

与列表扁平化相反。

给定一个列表和长度 n，返回长度为 n 的子列表的列表。

def sublist(lst, n):
    sub=[] ; result=[]
    for i in lst:
        sub+=[i]
        if len(sub)==n: result+=[sub] ; sub=[]
    if sub: result+=[sub]
    return result

举个例子：

如果列表是：

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

并且 n 是：

3

返回：

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

有没有更雄辩/简洁的方式？

顺便说一句，将列表附加到列表时首选什么（在上面的上下文中）：

list1+=[list2]

或者：

list1.append(list2)

考虑到（根据 Summerfeild 的“Python 3 编程”）它们是相同的？

谢谢。

Answer 1

这样的列表列表可以使用 列表理解 构建：

In [17]: seq=[1,2,3,4,5,6,7,8]
In [18]: [seq[i:i+3] for i in range(0,len(seq),3)]
Out[18]: [[1, 2, 3], [4, 5, 6], [7, 8]]

有还有 石斑鱼习语：

In [19]: import itertools
In [20]: list(itertools.izip_longest(*[iter(seq)]*3))
Out[20]: [(1, 2, 3), (4, 5, 6), (7, 8, None)]

但请注意，缺少的元素将用值 None 填充。 izip_longest 也可以采用 fillvalue 参数如果需要除“无”之外的其他内容。

---

list1+=[list2]——注意这次的括号——相当于list1.append(list2)。我写代码时最优先考虑的是可读性，
不是速度。因此，我会选择 list1.append(list2)。然而，可读性是主观的，并且可能很大程度上受到您所熟悉的习语的影响。

令人高兴的是，在这种情况下，可读性和速度似乎是一致的：

In [41]: %timeit list1=[1,2,3]; list1.append(list2)
1000000 loops, best of 3: 612 ns per loop

In [42]: %timeit list1=[1,2,3]; list1+=[list2]
1000000 loops, best of 3: 847 ns per loop

Answer 2

下面的内容怎么样（其中 x 是您的列表）：

 [x[i:i+3] for i in range(0, len(x), 3)]

对于 n!=3 来说，这很简单。

至于你的第二个问题，它们是等效的，所以我认为这是风格问题。但是，请确保您没有令人困惑<代码>附加和<代码>扩展。

Answer 3

您听说过 boltons 吗？

Boltons 是一组纯 Python 实用程序，其精神与标准库相同，但明显缺失

它具有您想要的内置功能，称为分块

from boltons import iterutils

iterutils.chunked([1,2,3,4,5,6,7,8], 3)

输出：

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

并且boltons 中更有吸引力的是它有分块作为迭代器，称为chunked_iter，因此您不需要存储整个事情都在记忆中。很整洁，对吧？

Answer 4

该函数可以采用任何类型的可迭代（不仅是已知长度的序列）：

import itertools

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

print(list(grouper(3, [1,2,3,4,5,6,7,8,9,10])))
# [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]

Answer 5

我认为这个 split 函数可以满足您的需求（尽管它适用于任何迭代器而不仅仅是列表）：

from itertools import islice

def take(n, it):
    "Return first n items of the iterable as a list"
    return list(islice(it, n))

def split(it, size):
    it = iter(it)
    size = int(size)
    ret = take(size, it)
    while ret:
        yield ret
        ret = take(size, it)

编辑：关于您的旁白，我总是使用 list.append(blah)，因为它对我来说更惯用，但是我相信它们在功能上是等效的。

Answer 6

对于某些特定情况，使用 numpy 包可能会很有用。在此包中，您有一个 reshape 例程：

import numpy as np
x = np.array([1,2,3,4,5,6])
np.reshape(x, (-1,3))

但是，如果它不是 n 的倍数，则此解决方案不会填充您的列表。

Answer 7

我知道，它看起来像 brainfuck，但它是有效的：

>>> a = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15]
>>> n = 3
>>> [i for j in [[a[t:t+n] for x in a[:1:t+1] if (t%n)==False] for t in range(len(a))] for i in j]
[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12], [13, 14, 15]]

>>> n = 4
>>> [i for j in [[a[t:t+n] for x in a[:1:t+1] if (t%n)==False] for t in range(len(a))] for i in j]
[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15]]