Python如何将列表分为给定长度的子列表()

将列表拆分为子列表的问题非常普遍, 但拆分给定长度的子列表的问题并不常见。给定一个列表列表和一个长度列表, 任务是将列表分成给定长度的子列表。
例子:

Input : Input = [1, 2, 3, 4, 5, 6, 7] length_to_split = [2, 1, 3, 1]Output: [[1, 2], [3], [4, 5, 6], [7]]

方法1:使用伊丽丝将列表分为给定长度的子列表是最优雅的方法。
# Python code to split a list # into sublists of given length.from itertools import islice# Input list initialization Input = [ 1 , 2 , 3 , 4 , 5 , 6 , 7 ]# list of length in which we have to split length_to_split = [ 2 , 1 , 3 , 1 ]# Using islice Inputt = iter ( Input ) Output = [ list (islice(Inputt, elem)) for elem in length_to_split]# Printing Output print ( "Initial list is:" , Input ) print ( "Split length list: " , length_to_split) print ( "List after splitting" , Output)

【Python如何将列表分为给定长度的子列表()】输出如下:
Initial list is: [1, 2, 3, 4, 5, 6, 7] Split length list:[2, 1, 3, 1] List after splitting [[1, 2], [3], [4, 5, 6], [7]]

方法2:
使用
压缩
是将列表拆分为给定长度的子列表的另一种方法。
# Python code to split a list into # sublists of given length. from itertools import accumulate# Input list initialization Input = [ 1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 9 , 10 ]# list of length in which we have to split length_to_split = [ 2 , 2 , 3 , 3 ]# Using islice Output = [ Input [x - y: x] for x, y in zip ( accumulate(length_to_split), length_to_split)]# Printing Output print ( "Initial list is:" , Input ) print ( "Split length list: " , length_to_split) print ( "List after splitting" , Output)

输出如下:
Initial list is: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] Split length list:[2, 2, 3, 3] List after splitting [[1, 2], [3, 4], [5, 6, 7], [8, 9, 10]]

首先, 你的面试准备可通过以下方式增强你的数据结构概念:Python DS课程。

    推荐阅读