段在Python列表(Segment a list in Python)

2019-09-21 22:34发布

我寻找一个蟒内置函数(或机构)到段列表成所需要的段长度(没有突变输入列表)。 这里是我已经有了代码:

>>> def split_list(list, seg_length):
...     inlist = list[:]
...     outlist = []
...     
...     while inlist:
...         outlist.append(inlist[0:seg_length])
...         inlist[0:seg_length] = []
...     
...     return outlist
... 
>>> alist = range(10)
>>> split_list(alist, 3)
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]

Answer 1:

您可以使用列表理解:

>>> seg_length = 3
>>> a = range(10)
>>> [a[x:x+seg_length] for x in range(0,len(a),seg_length)]
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]


Answer 2:

如何你需要使用的输出? 如果你只需要遍历它,你最好创建一个迭代器,一个得到您的群体:

def split_by(sequence, length):
    iterable = iter(sequence)
    def yield_length():
        for i in xrange(length):
             yield iterable.next()
    while True:
        res = list(yield_length())
        if not res:
            return
        yield res

用法示例:

>>> alist = range(10)
>>> list(split_by(alist, 3))
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]

它使用少得多的内存比试图建构在内存中的整个列表一次,如果你只在循环的结果,因为它一次只能构建一个子集:

>>> for subset in split_by(alist, 3):
...     print subset
...
[0, 1, 2]
[3, 4, 5]
[6, 7, 8]
[9]


Answer 3:

不一样的输出,我仍然认为石斑鱼功能是有帮助的:

from itertools import izip_longest
def grouper(iterable, n, fillvalue=None):
    args = [iter(iterable)] * n
    return izip_longest(*args, fillvalue=fillvalue)

为的python2.4和2.5不具有izip_longest:

from itertools import izip, chain, repeat
def grouper(iterable, n, padvalue=None):
    return izip(*[chain(iterable, repeat(padvalue, n-1))]*n)

一些演示代码和输出:

alist = range(10)
print list(grouper(alist, 3))

输出:[(0,1,2),(3,4,5),(6,7,8),(9,无,无)]



文章来源: Segment a list in Python