如何將子列表拆分爲子列表k次？ （蟒蛇）

Question

我知道我可以使用拆分列表分成大小相同的子列表：

segment = len(list)//k 
sub_lists = [list[i:i+segment] for i in range(0, len(list), segment)] 

但是我不知道怎麼的長度K^M列表分成子列表，再往子列表，直到每個子列表有1

例如長度：

k = 2 
list = [1, 2, 3, 4, 5, 6, 7, 8] 
list = [[1, 2, 3, 4], [5, 6, 7, 8]] 
list = [[[1, 2], [3, 4]], [[5, 6], [7, 8]]] 
list = [[[[1], [2]], [[3], [4]]], [[[5], [6]], [[7], [8]]]] 

每當我試圖循環這個我在海里獲得並列，有沒有捷徑？

Answer 1

這基本上是你的代碼：

def split_list(input_list, segments): 
    if len(input_list) == 1: 
     return input_list 
    segment_length = len(input_list) // segments 
    return [split_list(input_list[i:i+segment_length], segments) 
      for i in range(0, len(input_list), segment_length)] 

>>> split_list([1, 2, 3, 4, 5, 6, 7, 8], 2) 
[[[[1], [2]], [[3], [4]]], [[[5], [6]], [[7], [8]]]] 

Answer 2

def sub_k_list(a, k): 
    p = len(a) // k 
    return a if not p else [sub_k_list(a[:p], k), sub_k_list(a[p:], k)] 

k = 2 
a = [1, 2, 3, 4, 5, 6, 7, 8] 

print(sub_k_list(a, k)) 

結果：

[[[[1], [2]], [[3], [4]]], [[[5], [6]], [[7], [8]]]] 

編輯：刪除if ...

def sub_k_list(a, k): 
    p = len(a) // k 
    return a * (not p) or [sub_k_list(a[:p], k), sub_k_list(a[p:], k)]