python一行代码实现将一个list均分

需求描述：将一个list按长度n均分

例如：

1. 输入list为： list_data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
2. 均分长度n为： num=3
3. 预期输出结果：[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]

[list_data[i*num:(i+1)*num] for i in range(int(len(list_data)/num) + 1) if list_data[i*num:(i+1)*num]]

代码如下：

# 需求
# 需求描述：将一个list按长度n均分
#
# 例如：
#
# 输入list为：
list_data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
# 均分长度n为：
num = 3

# 预期输出结果：[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]

# 1. 一行代码
result = [list_data[i*num:(i+1)*num] for i in range(int(len(list_data)/num) + 1) if list_data[i*num:(i+1)*num]]

# 2. 一行函数版
def list_equally_split(list_data, num): return [list_data[i*num:(i+1)*num] for i in range(int(len(list_data)/num) + 1) if list_data[i*num:(i+1)*num]]