Python实现带偏移量的双列表Zip并保留首尾剩余元素

要实现需求中两个列表按指定元素对齐、拆分出左侧剩余、中间对齐段和右侧剩余的功能，可以通过计算对齐偏移量，结合列表切片和交替合并来完成，以下是具体实现：

Python 代码实现 #

def align_lists(list1, list2, index1, index2):
    list2_len = len(list2)
    # 计算list1中对齐片段的起始和结束索引
    align_start = index1 - index2
    align_end = align_start + list2_len
    
    # 提取左侧剩余元素
    left = list1[:align_start] if align_start > 0 else []
    # 提取右侧剩余元素
    right = list1[align_end:] if align_end < len(list1) else []
    
    # 截取list1中用于对齐的片段
    list1_segment = list1[max(align_start, 0) : min(align_end, len(list1))]
    # 截取list2中对应对齐的片段
    list2_offset = max(-align_start, 0)
    list2_segment = list2[list2_offset : list2_offset + len(list1_segment)]
    
    # 交替合并对齐片段
    paired = []
    for elem1, elem2 in zip(list1_segment, list2_segment):
        paired.append(elem1)
        paired.append(elem2)
    
    # 处理list2中无法配对的前置/后置元素
    list2_left = list2[:list2_offset] if list2_offset > 0 else []
    list2_right = list2[list2_offset + len(list1_segment):] if (list2_offset + len(list1_segment)) < list2_len else []
    
    zipped_middle = list2_left + paired + list2_right
    
    return left, zipped_middle, right

# 测试示例
if __name__ == "__main__":
    list1 = [1, 2, 3, 4, 5, 6]
    list2 = ["a", "b", "c"]
    index1 = 2  # list1中要对齐的元素索引
    index2 = 1  # list2中要对齐的元素索引
    
    left, zipped_middle, right = align_lists(list1, list2, index1, index2)
    final_list = left + zipped_middle + right
    
    print("left:", left)
    print("zipped_middle:", zipped_middle)
    print("right:", right)
    print("final_list:", final_list)

代码运行结果 #

left: [1]
zipped_middle: [2, 'a', 3, 'b', 4, 'c']
right: [5, 6]
final_list: [1, 2, 'a', 3, 'b', 4, 'c', 5, 6]

逻辑说明 #

1. 计算对齐范围：通过index1 - index2得到list1中对齐片段的起始位置，起始位置加上list2的长度得到对齐片段的结束位置。
2. 提取剩余元素：
   • 左侧剩余：list1中对齐起始位置之前的元素，若起始位置为负则左侧剩余为空。
   • 右侧剩余：list1中对齐结束位置之后的元素，若结束位置超出list1长度则右侧剩余为空。
3. 构建中间对齐段：
   • 截取list1和list2中可配对的片段，交替合并这两个片段。
   • 若list2有无法配对的前置元素（当对齐起始位置为负时）或后置元素（当对齐结束位置超出list1长度时），将这些元素分别添加到合并后的片段前后，形成最终的中间对齐段。

该方法适配任意长度的列表，无论指定的对齐索引在列表中的位置如何，都能保证指定元素在中间对齐段中相邻，同时保留所有原始元素。

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