Python中移除列表a与列表b匹配值的代码问题排查

原代码的逻辑漏洞 #

• 循环过滤的基准错误：每次发现val在b中时，你始终基于**原始列表a**执行过滤，而非之前已经处理过的new_array。以测试用例a=[1,2,3], b=[1,2]为例：
  • 遍历到1时，过滤原始a得到[2,3]并赋值给new_array
  • 遍历到2时，再次过滤原始a得到[1,3]，直接覆盖了之前的结果
  • 最终返回[1,3]，与预期的[3]不符
• 无匹配元素时返回空列表：如果a中所有元素都不在b里，循环内不会执行任何赋值操作，最终返回初始的空列表[]，完全不符合需求
• 空b的处理冗余且有风险：将b为空的判断放在循环内部，每次遍历都重复检查；且直接返回a的引用（而非副本），外部修改原列表a会影响返回结果

修正后的实现 #

基础直观版 #

def array_diff(a, b):
    new_array = []
    for val in a:
        if val not in b:
            new_array.append(val)
    return new_array

高效优化版 #

当b元素较多时，将其转为集合可把val in b的时间复杂度从O(n)降至O(1)：

def array_diff(a, b):
    b_set = set(b)
    return [val for val in a if val not in b_set]

逻辑说明 #

两种实现均严格对应需求：遍历a的每个元素，仅保留不在b中的元素。测试用例a=[1,2,3], b=[1,2]会返回[3]，符合预期。

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