Python实现列表交错合并的简洁方法及通用多场景适配

首先针对你提到的两个列表的场景，分两种情况给出解决方案：

情况1：后缀元素需要从原列表生成（如a_1、b_2这类带递增序号的） #

你期望的结果是原元素和对应递增序号的后缀元素交错排列，用嵌套列表推导式就能轻松实现：

l = ['a', 'b', 'c']
out_l = [item for pair in zip(l, (f"{elem}_{idx+1}" for idx, elem in enumerate(l))) for item in pair]
print(out_l)  # 输出: ['a', 'a_1', 'b', 'b_2', 'c', 'c_3']

解释： #

1. 用enumerate(l)遍历原列表，同时拿到元素和它的索引（从0开始），通过生成器表达式生成带递增序号的后缀元素（idx+1让序号从1开始）；
2. zip(l, 生成器)把原元素和对应的后缀元素配对成元组，比如('a', 'a_1')；
3. 嵌套列表推导式把每个元组里的元素依次展开，得到最终的交错列表。

情况2：给定两个任意列表（后缀列表并非由原列表派生） #

如果是两个现成的列表需要交错合并，代码更简洁：

l = ['a', 'b', 'c']
l2 = ['a_1', 'b_1', 'c_1']
out_l = [item for pair in zip(l, l2) for item in pair]
print(out_l)  # 输出: ['a', 'a_1', 'b', 'b_1', 'c', 'c_1']

扩展：多个列表的通用交错合并方案 #

如果要支持任意数量的列表（不管是否由原列表派生），可以用两种方式实现：

方法1：用itertools.chain.from_iterable（简洁高效） #

借助标准库itertools的工具，能快速处理多个列表的交错：

import itertools

l1 = ['a', 'b', 'c']
l2 = ['x', 'y', 'z']
l3 = ['1', '2', '3']

# 按位置交错合并，以最短列表长度为准
out_l = list(itertools.chain.from_iterable(zip(l1, l2, l3)))
print(out_l)  # 输出: ['a', 'x', '1', 'b', 'y', '2', 'c', 'z', '3']

如果列表长度不一致，想要保留所有元素，可以用itertools.zip_longest，再过滤掉填充的None：

import itertools

l1 = ['a', 'b', 'c', 'd']
l2 = ['x', 'y']
l3 = ['1', '2', '3']

# 用zip_longest处理不等长列表，fillvalue指定填充值
merged_groups = itertools.zip_longest(l1, l2, l3, fillvalue=None)
out_l = [item for group in merged_groups for item in group if item is not None]
print(out_l)  # 输出: ['a', 'x', '1', 'b', 'y', '2', 'c', '3', 'd']

方法2：自定义函数（纯列表推导式，无需依赖库） #

如果你不想用标准库，也可以写一个通用函数，用*args接收任意数量的列表：

def interleave_lists(*lists):
    return [item for group in zip(*lists) for item in group]

# 使用示例
l1 = ['a', 'b']
l2 = ['x', 'y']
l3 = ['1', '2']
print(interleave_lists(l1, l2, l3))  # 输出: ['a', 'x', '1', 'b', 'y', '2']

这个函数同样以最短列表的长度为合并基准，如果要处理不等长列表，可以结合zip_longest的思路修改。

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