Python：如何按给定子列表索引序列逐步移除嵌套列表元素

问题根源 #

你的代码中变量u始终固定为0，导致每次循环都基于初始的J[0]（即[2, 6, 9, 10]）进行处理，而不是用上一次生成的新列表。这就是为什么第二次输出的是从原列表移除索引1的元素，而非从第一次结果[6,9]中移除索引1的元素。

修正后的代码 #

写法一：更新索引变量u

J=[[2, 6, 9, 10]]
index=[[0,3],[1]]
u=0
for v in range(len(index)): 
    new_J = [j for i, j in enumerate(J[u]) if i not in index[v]]
    J.append(new_J)
    u += 1  # 每次循环后更新u，指向刚追加的新列表
print(J)

写法二：直接取列表最后一个元素

更简洁的方式是直接使用J[-1]获取最新生成的列表，无需维护额外的索引变量：

J=[[2, 6, 9, 10]]
index=[[0,3],[1]]
for idx_list in index: 
    new_J = [j for i, j in enumerate(J[-1]) if i not in idx_list]
    J.append(new_J)
print(J)

输出结果 #

两种写法都会得到你期望的输出：

[[2, 6, 9, 10], [6, 9], [6]]

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