创建子列表

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