Python中移除列表内子列表指定元素报错问题求助

我有一个包含多个子列表的列表A，希望从每个子列表中移除指定元素：从A[0]移除元素0，从A[1]移除元素1，依此类推。但运行代码时出现错误，以下是我的代码、报错信息及期望输出。

我的代码 #

A=[[0, 2, 3, 5], [1, 3, 4, 6], [0, 2, 3, 5], [0, 1, 2, 3, 4, 5, 6]]

for i in range(0,len(A)): 
    A[i].remove(A[i]==i)
print(A)

报错信息 #

in <module>
    A[i].remove(A[i]==i)

ValueError: list.remove(x): x not in list

期望输出 #

[[2, 3, 5], [3, 4, 6], [0, 3, 5], [0, 1, 2, 4, 5, 6]]

错误原因 #

问题出在A[i].remove(A[i]==i)这行代码：A[i]==i是一个布尔判断，返回的是True或False，但你实际要移除的是数字i，列表里不存在布尔值，所以触发了「元素不存在」的ValueError。

修正方案 #

方案1：直接修正目标元素 #

把A[i]==i换成要移除的数字i即可：

A = [[0, 2, 3, 5], [1, 3, 4, 6], [0, 2, 3, 5], [0, 1, 2, 3, 4, 5, 6]]

for i in range(len(A)): 
    A[i].remove(i)
print(A)

注意：如果某个子列表中不存在i，remove()仍会报错，适合确定每个子列表都包含目标元素的场景。

方案2：先判断元素存在再移除 #

增加存在性检查，避免报错：

A = [[0, 2, 3, 5], [1, 3, 4, 6], [0, 2, 3, 5], [0, 1, 2, 3, 4, 5, 6]]

for i in range(len(A)):
    if i in A[i]:
        A[i].remove(i)
print(A)

方案3：用列表推导式生成新列表（更安全简洁） #

通过列表推导式过滤掉目标元素，生成新列表，不修改原列表，且无需担心元素不存在的问题：

A = [[0, 2, 3, 5], [1, 3, 4, 6], [0, 2, 3, 5], [0, 1, 2, 3, 4, 5, 6]]

A = [ [num for num in sublist if num != i] for i, sublist in enumerate(A) ]
print(A)

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