今天写一个任务调度分发模块的时候遇上一个需求就是,需要将输入的任务列表分割进而创建子任务列表,每个子任务数量相同,这就是一个均分的问题,很简单,不多说,使用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]]yishuihancheng
来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/69927062/viewspace-3148775/ ,如需转载,请注明出处,否则将追究法律责任。