Python中实现普通列表与嵌套列表匹配并移除匹配元素的方法

你的问题核心在于原代码仅检查元素是否在列表B的顶层元素中，而B里的'red'处于子列表内，无法被item in B检测到，自然也无法完成移除操作。以下是修改后的实现方案：

实现代码 #

def remove_nested_item(lst, item):
    for i, elem in enumerate(lst):
        if elem == item:
            lst.pop(i)
            return True
        elif isinstance(elem, list):
            if remove_nested_item(elem, item):
                # 子列表移除元素后为空时，同步删除该空列表
                if not elem:
                    lst.pop(i)
                return True
    return False

A = ['yes', 'red', 'car']
B = ['yes', ['yellow', 'red'], 'car']

for item in A:
    if remove_nested_item(B, item):
        print(f"{item} is a correct answer.")
    else:
        print(f"{item} is a wrong answer.")

# 验证最终B的状态
print("最终B列表:", B)

代码说明 #

1. 递归辅助函数：remove_nested_item会逐层遍历列表（包括嵌套子列表）：
   • 若找到匹配元素，直接移除并返回True
   • 遇到子列表时递归进入查找，若子列表内元素被移除后变为空，会将该空列表从父列表中删除
2. 主逻辑：遍历列表A的每个元素，调用辅助函数完成查找与移除，匹配成功则打印对应提示

运行后会输出3条X is a correct answer.，最终B列表会变为空，完全符合你的需求。

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