如何基于部分元素匹配合并两个嵌套列表?
嵌套列表按交集合并的解决方案
针对你需要将嵌套列表A和B中存在共同元素的子列表合并的需求,这里提供Python实现方案:
核心逻辑
- 遍历A中的每个子列表
- 对每个A的子列表,筛选出B中所有与之存在共同元素的子列表
- 将匹配的A子列表和B子列表合并(拼接后排序,保留重复元素)
- 把每个A子列表对应的所有合并结果组成子列表,最终生成嵌套列表C
代码实现
# 示例输入 A = [ [3, 10], [4, 25, 31], [2, 10] ] B = [ [2, 6, 9, 10], [15, 25, 26], [2, 6, 8, 10], [16, 20, 21] ] # 生成结果列表C C = [] for a_sub in A: current_merges = [] a_elements = set(a_sub) for b_sub in B: # 检查两个子列表是否有共同元素 if a_elements & set(b_sub): # 合并并排序(匹配示例中的结果格式) combined = sorted(a_sub + b_sub) current_merges.append(combined) C.append(current_merges) # 输出结果 print(C)
输出结果
[ [[2, 3, 6, 9, 10, 10], [2, 3, 6, 8, 10, 10]], [[4, 15, 25, 25, 26, 31]], [[2, 2, 6, 9, 10, 10], [2, 2, 6, 8, 10, 10]] ]
说明
- 用集合判断交集能大幅提升效率,适合处理较大规模的嵌套列表(比如A包含15+子列表的场景)
- 合并时保留重复元素并排序,完全匹配你给出的示例逻辑(示例中第三个子列表的第一个结果可能存在笔误,实际合并后会保留两个2)
内容的提问来源于stack exchange,提问作者user12269799
相关产品推荐
相关产品推荐

