Python实现将一个长度为n的列表划分，每个子列表中包含m个元素

今天写一个任务调度分发模块的时候遇上一个需求就是，需要将输入的任务列表分割进而创建子任务列表，每个子任务数量相同，这就是一个均分的问题，很简单，不多说，使用Python来做很简单，下面是两种做法的简单实现：

def test1(one_data_list,colnum=3):
 '''
 将一维的列表转化为矩阵形式
 '''
 res_list=[]
 for i in range(0,len(one_data_list),colnum):
 res_list.append(one_data_list[i:i+colnum])
 return res_list

def test2(one_list,c=3):
 '''
 将一个长度为n的列表划分 ，每个子列表中包含m个元素
 '''
 return [one_list[i:i+c] for i in range(len(one_list)) if i%c==0]

简单测试一下：

one_data_list=[1,2,3,4,5,6,7,8,9]
print test1(one_data_list,colnum=3)
print test2(one_data_list,c=3)

结果如下：

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