将扁平列表转换为指定长度子列表的交错填充方法

def convert_list_to_interleaved_sublists(big_list, sublist_lengths):
    """
    将扁平列表转换为指定长度的子列表列表，采用交错式填充。

    Args:
        big_list (list): 待转换的原始扁平列表。
        sublist_lengths (list): 一个整数列表，定义了每个子列表的期望长度。

    Returns:
        list: 转换后的子列表列表。

    Raises:
        AssertionError: 如果所有子列表长度之和不等于原始列表的长度。
    """

    # 1. 验证输入：确保总元素数量匹配
    if sum(sublist_lengths) != len(big_list):
        raise AssertionError("所有子列表的期望长度之和必须等于原始列表的长度。")

    # 2. 初始化：创建空的子列表列表
    # lol (list_of_lists) 将存储最终结果
    lol = [[] for _ in sublist_lengths]

    # 3. 填充逻辑：交错式分配元素
    current_sublist_index = -1 # 用于循环遍历子列表的索引

    for element in big_list:
        while True:
            # 移动到下一个子列表
            current_sublist_index += 1
            # 如果索引超出子列表列表的范围，则从头开始循环
            if current_sublist_index == len(lol):
                current_sublist_index = 0

            # 检查当前子列表是否已达到其预定长度
            if len(lol[current_sublist_index]) < sublist_lengths[current_sublist_index]:
                # 如果未达到，则将当前元素添加到该子列表
                lol[current_sublist_index].append(element)
                break # 元素已添加，跳出内层while循环，处理下一个big_list元素
            # 如果已达到，则继续内层while循环，尝试下一个子列表

    return lol

# 示例使用
big_list = [1, 2, 3, 4, 5, 6, 7, 8]
sublist_lengths = [1, 2, 3, 2]

try:
    result_list_of_lists = convert_list_to_interleaved_sublists(big_list, sublist_lengths)
    print(f"原始列表: {big_list}")
    print(f"子列表长度定义: {sublist_lengths}")
    print(f"转换结果: {result_list_of_lists}")
except AssertionError as e:
    print(f"错误: {e}")

# 另一个示例
big_list_2 = ['a', 'b', 'c', 'd', 'e', 'f']
sublist_lengths_2 = [2, 1, 3]
try:
    result_list_of_lists_2 = convert_list_to_interleaved_sublists(big_list_2, sublist_lengths_2)
    print(f"\n原始列表: {big_list_2}")
    print(f"子列表长度定义: {sublist_lengths_2}")
    print(f"转换结果: {result_list_of_lists_2}")
except AssertionError as e:
    print(f"错误: {e}")

原始列表: [1, 2, 3, 4, 5, 6, 7, 8]
子列表长度定义: [1, 2, 3, 2]
转换结果: [[1], [2, 5], [3, 6, 8], [4, 7]]

原始列表: ['a', 'b', 'c', 'd', 'e', 'f']
子列表长度定义: [2, 1, 3]
转换结果: [['a', 'd'], ['b'], ['c', 'e', 'f']]