數組（list）分組、分段

對一個list進行分組，要求控制每組中的元素個數：

1.使用切片分組：

lst = [1, 2, 3, 4, 5, 6, 7, 8, 9, 8, 7, 6, 5, 4, 3, 2, 1] #lst可為空，最后返回值也為空
num=3 #定義每組包含的元素個數
for i in range(0,len(lst),num):
        print lst[i:i+num]

返回：
[1, 2, 3]
[4, 5, 6]
[7, 8, 9]
[8, 7, 6]
[5, 4, 3]
[2, 1]

2.使用迭代方式：

lst = [1, 2, 3, 4, 5, 6, 7, 8, 9, 8, 7, 6, 5, 4, 3, 2, 1] #lst可為空

num=3 #只要不為0即可，大於lst長度也可以
def lstg(num,lst):
    for i in range(0,len(lst),num):
        yield lst[i:i+num]
lstgs = lstg(num,lst)      

for i in lstgs:
    print i
返回：
[1, 2, 3]
[4, 5, 6]
[7, 8, 9]
[8, 7, 6]
[5, 4, 3]
[2, 1]

 

原始笨方法：

# -*- coding: UTF-8 -*-
lst = [1, 2, 3, 4, 5, 6, 7, 8, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0]
num=4

def lstg(num,lst):
#定義每段的個數num
    l = len(lst)
    #取分成幾組
    g = l/num
    #判斷是否有剩余的數
    last = l%num

    lstn = []
    for i in range(g):
        i=i+1
        n=i*num
        m=n-num    
        lstn.append(lst[m:n])

    if  last <> 0:
        lstn.append(lst[-last:])
    return lstn

for i in lstg(num,lst):
    print i

 

print divmod(13,3) #返回(4, 1)