需求描述:將一個list按長度n均分
例如:
[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]]