如何在Python中删除嵌套列表中的元素？

针对你要删除前两个子列表里的[0.0, inf]元素，这里提供两种实用方法：

方法一：直接使用.remove()（适合确定子列表中仅存在一个目标元素的场景） #

先定义目标元素，然后针对前两个子列表调用.remove()方法即可：

# 假设你的嵌套列表名为nested_list
target_element = [0.0, float('inf')]

# 直接删除第一个子列表中的目标元素
nested_list[0].remove(target_element)
# 删除第二个子列表中的目标元素
nested_list[1].remove(target_element)

如果担心子列表中可能不存在目标元素导致报错，可以先做存在性判断：

target_element = [0.0, float('inf')]
# 仅遍历前两个子列表
for sub_list in nested_list[:2]:
    if target_element in sub_list:
        sub_list.remove(target_element)

方法二：列表推导式过滤（适合需要删除所有匹配元素或不想修改原列表的场景） #

通过列表推导式生成过滤后的新子列表，替换原嵌套列表的前两个子项：

target_element = [0.0, float('inf')]
# 遍历前两个子列表的索引
for idx in range(2):
    # 过滤掉子列表中的目标元素，生成新列表
    nested_list[idx] = [item for item in nested_list[idx] if item != target_element]

补充说明 #

• .remove()只会删除子列表中第一个匹配的元素，如果子列表中有多个[0.0, inf]，推荐用列表推导式。
• 若不想修改原列表，而是生成新的嵌套列表，可参考以下写法：

new_nested_list = []
target_element = [0.0, float('inf')]
for idx, sub_list in enumerate(nested_list):
    if idx < 2:
        new_sub = [item for item in sub_list if item != target_element]
        new_nested_list.append(new_sub)
    else:
        new_nested_list.append(sub_list.copy())

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