如何将子列表分成k个子列表？（Python）

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]]]]

每当我尝试循环这个时，我就会陷入困境，有没有捷径？

- Anna

---

你想把每个列表分成 k 个子列表，对吗？ - Finwood

是的，基本上是这样。如果不太清楚，我很抱歉。 - Anna

只是出于好奇，这是一项学术练习，还是你有某个目标在心？ - Wayne Werner

这是我需要完成的一个学术练习中排序程序的一部分。 - Anna

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]]]]

- Finwood

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)]

- mementum