如何按有序顺序遍历多个不同已排序列表的所有元素？

你面临的场景是多个已排序的列表，需要按有序顺序遍历所有元素，同时要保留原列表，且元素类型不同需区别处理。原代码存在扩展性差（新增列表需大幅修改逻辑）、效率低（pop(0)是O(n)操作）、代码冗余的问题，下面给出两种更高效优雅的解决方案：

方案一：使用标准库heapq.merge（推荐） #

Python标准库的heapq.merge是专门为合并多个已排序可迭代对象设计的，它采用惰性迭代（不会一次性生成完整的合并列表，内存友好），时间复杂度为O(n log k)（n是总元素数，k是列表数量），非常高效。因为你需要保留原列表，只需提前复制每个列表即可。

from heapq import merge
from random import randint

# 生成测试数据
a: list = []
b: list = []
c: list = []
list_of_lists = [a, b, c]
for i in range(10):
    list_of_lists[randint(0, 2)].append(i)

print("原列表:", a, b, c)

# 复制所有列表，避免修改原数据
list_copies = [lst.copy() for lst in list_of_lists]

# 合并遍历，同时根据元素类型做区别处理
for elem in merge(*list_copies):
    # 这里根据实际元素类型编写处理逻辑
    if isinstance(elem, int):
        print(f"处理整数元素: {elem}")
    # elif isinstance(elem, str):
    #     print(f"处理字符串元素: {elem}")
    # 其他类型的处理逻辑...

优点： #

• 代码简洁，一行完成合并逻辑
• 惰性迭代，内存占用低，适合处理大列表
• 扩展性强：新增列表只需添加到list_of_lists中，无需修改遍历逻辑

方案二：自定义迭代器合并（适合需要额外上下文的场景） #

如果需要追踪元素来自哪个原列表，或者需要更精细的控制逻辑，可以用迭代器来实现，避免pop(0)的低效操作（列表pop(0)每次操作会移动所有元素，时间复杂度O(n)），改用迭代器的next()方法（O(1)操作）。

from random import randint

# 生成测试数据
a: list = []
b: list = []
c: list = []
list_of_lists = [a, b, c]
for i in range(10):
    list_of_lists[randint(0, 2)].append(i)

print("原列表:", a, b, c)

# 为每个复制后的列表创建迭代器
iters = [iter(lst.copy()) for lst in list_of_lists]
# 初始化每个迭代器的当前元素
current_elements = []
for it in iters:
    try:
        current_elements.append(next(it))
    except StopIteration:
        current_elements.append(None)

while any(item is not None for item in current_elements):
    # 筛选出当前有效的非None元素及其索引
    valid_pairs = [(val, idx) for idx, val in enumerate(current_elements) if val is not None]
    if not valid_pairs:
        break
    # 找到当前最小的元素
    min_val, min_idx = min(valid_pairs)
    
    # 根据元素类型区别处理，同时可获取元素来源列表的索引
    if isinstance(min_val, int):
        print(f"来自列表{min_idx}的整数元素: {min_val}")
    # 其他类型处理逻辑...
    
    # 从对应迭代器取下一个元素
    try:
        current_elements[min_idx] = next(iters[min_idx])
    except StopIteration:
        current_elements[min_idx] = None

优点： #

• 效率更高，避免了pop(0)的性能损耗
• 可追踪元素来源，适合需要上下文的业务场景
• 扩展性好，新增列表只需添加到list_of_lists，无需修改核心逻辑

内容的提问来源于stack exchange，提问作者Andreas Schuldei