在Python中高效地分组相邻列表项

Question

我想用固定数字对相邻列表项进行分组。如

a = [1, 2, 3, 4, 5, 6]

的groupby(a, 2)结果是[(1, 2), (3, 4), (5, 6)]，

的groupby(a, 3)结果是[(1, 2, 3), (4, 5, 6)]

这里是我的解决方案

def groupby(a, l): 
    ret = [] 
    for idx in range(len(a)/l): 
     ret.append(tuple(a[idx*l:(idx+1)*l:])) 
    return ret 

有没有更有效的方法来实现它？

Answer 1

是的，您打算使用zip。

编辑：固定格式

具体来说，

def groupby(a, l): 
    ret = [] 
    n = len(a) // l 
    it = len(a) // n 
    return list(zip(*(a[i * it:i * it + it] for i in range(n)))) 

Answer 2

我们可以通过itertools.izip_longest

做请检查下面的链接：

Alternative way to split a list into groups of n

Answer 3

>>> a = [1, 2, 3, 4, 5, 6] 
>>> groupby = lambda l, n: [tuple(l[i:i+n]) for i in range(0, len(l), n)] 
>>> groupby(a, 3) 
[(1, 2, 3), (4, 5, 6)] 
>>> groupby(a, 2) 
[(1, 2), (3, 4), (5, 6)] 
>>>