Python如何优雅实现嵌套列表独有元素提取与去重
简洁实现方案
用集合做元素存在性判定,替代多层嵌套for循环的手动判断,代码更简洁且运行效率更高,完全匹配你给出的示例规则:
- 第一步先展平嵌套列表,快速筛选出仅在list1中出现、未在list2中出现的元素,按list1中的原有顺序去重后作为新子列表追加到list2末尾
- 第二步跨子列表去重核心逻辑为倒序遍历子列表,用集合记录已经在后续子列表出现过的元素,仅保留当前子列表中未在后续出现过的元素,最后将结果反转回原顺序即可,刚好适配示例中「重复元素仅保留在最后一次出现的子列表中」的规则。
完整代码
# 初始化原始列表 list1 = [[1,2,3],[2,5],[6,7,4]] list2 = [[1,3],[2],[6,4],[9,0,3]] # 步骤1:计算差集并追加到list2 set2 = set(num for sub in list2 for num in sub) new_list = [] seen_diff = set() # 按list1的元素遍历顺序取差集,保证结果顺序和示例一致 for num in (n for sub in list1 for n in sub): if num not in set2 and num not in seen_diff: seen_diff.add(num) new_list.append(num) list2.append(new_list) # 封装跨子列表去重函数 def deduplicate_nested(nested_list): seen = set() temp = [] # 倒序遍历所有子列表 for sub in reversed(nested_list): new_sub = [] for num in sub: if num not in seen: seen.add(num) new_sub.append(num) temp.append(new_sub) # 反转回原始子列表顺序 return temp[::-1] list1 = deduplicate_nested(list1) list2 = deduplicate_nested(list2)
运行结果
代码执行后得到的结果和示例完全一致:
list1取值为[[1, 3], [2, 5], [6, 7, 4]]list2取值为[[1], [2], [6, 4], [9, 0, 3], [5, 7]]
如果后续需要调整去重规则为「重复元素保留在第一次出现的子列表中」,只需要把去重函数里的倒序遍历改为正序遍历,去掉最后反转列表的逻辑即可,调整成本很低。
内容的提问来源于stack exchange,提问作者DANY
相关产品推荐
相关产品推荐

