如何将子列表拆分为子列表k次? (Python)

前端之家收集整理的这篇文章主要介绍了如何将子列表拆分为子列表k次? (Python)前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我知道我可以使用以下方法将列表拆分为相同大小的子列表:
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,4],[5,8]]
list = [[[1,2],[3,4]],[[5,6],[7,8]]]
list = [[[[1],[2]],[[3],[4]]],[[[5],[6]],[[7],[8]]]]

每当我试图循环这个时,我就会被打结,是否有捷径?

解决方法

这基本上就是你的代码
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,8],2)
[[[[1],[8]]]]

猜你在找的Python相关文章