python 一行代码实现对列表的快速等分

本文实现了两种方式,定义如下函数chunk1和chunk2,参数l为要划分的列表,chunk_size为每个划分的大小。可以看到,函数中只有一行代码,即可实现该功能。不过显然方式二会简洁一些。

import math

def chunk1(l,chunk_size):
    return list(map(lambda x:l[x*chunk_size:(x+1)*chunk_size],range(0,math.ceil(len(l)/chunk_size)))


def chunk2(l,chunk_size):
    return [l[x:x + chunk_size] for x in range(0,len(l),chunk_size)]


l=[0,1,2,3,4,5,6,7,8,9]
chunk1(l,3)
[out]:
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]

chunk2(l,4)
[out]:
[[0, 1, 2, 3], [4, 5, 6, 7], [8, 9]]

猜你喜欢

转载自blog.csdn.net/S_o_l_o_n/article/details/107041093