构建列表金字塔结构：使用itertools的优雅方法

from itertools import count, islice

def build_pyramid(source_iterable):
    """
    将一个可迭代对象转换为金字塔结构（列表的列表），
    其中每个子列表的元素数量依次递增。

    Args:
        source_iterable: 任意可迭代对象，例如列表、字典的keys()视图或生成器。

    Yields:
        list: 金字塔结构中的一个子列表（一层）。
    """
    # 确保输入是一个迭代器，以便可以逐次消耗
    diter = iter(source_iterable)

    # count(1) 生成无限序列 1, 2, 3, ... 作为每一层的期望长度
    for i in count(1):
        # islice 尝试从 diter 中取出 i 个元素
        current_row = list(islice(diter, i))

        # 检查是否成功取出了 i 个元素
        # 如果 len(current_row) == i，说明这一层是完整的
        if len(current_row) == i:
            yield current_row
        else:
            # 如果取出的元素数量少于 i，说明源迭代器已耗尽，
            # 此时如果 current_row 不为空，也应作为最后一层返回
            if current_row:
                yield current_row
            return # 源迭代器已耗尽，停止生成

# 示例1：使用一个简单的列表
numbers_list = list(range(1, 11)) # [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
pyramid_structure_1 = list(build_pyramid(numbers_list))
print(f"从列表 {numbers_list} 构建的金字塔: {pyramid_structure_1}")
# 预期输出: 从列表 [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 构建的金字塔: [[1], [2, 3], [4, 5, 6], [7, 8, 9, 10]]

# 示例2：使用字典的键（通常需要先排序）
encoded_message = {1: 'a', 2: 'b', 3: 'c', 4: 'd', 5: 'e', 6: 'f', 7: 'g', 8: 'h', 9: 'i', 10: 'j'}
# 为了保证顺序，通常需要对字典键进行排序
sorted_keys_iterator = iter(sorted(encoded_message.keys()))
pyramid_structure_2 = list(build_pyramid(sorted_keys_iterator))
print(f"从字典键构建的金字塔: {pyramid_structure_2}")
# 预期输出: 从字典键构建的金字塔: [[1], [2, 3], [4, 5, 6], [7, 8, 9, 10]]

# 示例3：输入列表元素不足以形成完整金字塔的情况
short_list = [1, 2, 3, 4, 5]
pyramid_structure_3 = list(build_pyramid(short_list))
print(f"从短列表 {short_list} 构建的金字塔: {pyramid_structure_3}")
# 预期输出: 从短列表 [1, 2, 3, 4, 5] 构建的金字塔: [[1], [2, 3], [4, 5]]