ITPub博客

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

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

作者:u_509a2a8fdb46 时间:2019-08-30 11:07:37 0 删除 编辑

今天写一个任务调度分发模块的时候遇上一个需求就是,需要将输入的任务列表分割进而创建子任务列表,每个子任务数量相同,这就是一个均分的问题,很简单,不多说,使用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/ ,如需转载,请注明出处,否则将追究法律责任。

请登录后发表评论 登录
全部评论

注册时间:2019-08-08

  • 博文量
    210
  • 访问量
    38