今天写一个任务调度分发模块的时候遇上一个需求就是,需要将输入的任务列表分割进而创建子任务列表,每个子任务数量相同,这就是一个均分的问题,很简单,不多说,使用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]]