在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)] 
>>>