Python 3中如何按索引列表删除嵌套列表元素

嗨，这个问题我之前也碰到过，咱们一步步来理清楚：

首先，你遇到的TypeError原因很明确：A里的每个元素都是列表，而del W[t][x]要求x必须是单个整数或者切片，你直接把整个列表传进去，Python肯定不知道要怎么处理啦。

问题核心 #

你的需求是：对于每个位置t，要删除W[t]中所有在A[t]里列出的索引对应的元素。但这里有个坑：如果直接按升序删除索引，前面的元素被删掉后，后面元素的索引会往前移位，导致你删错元素。

举个例子：W[1] = [1,2,3]，A[1] = [0,1]，如果先删索引0，W[1]变成[2,3]，这时候再删索引1，删掉的是3，但你原本想删的是原来的索引1对应的2，结果就错了。

解决方案1：修改原列表（降序删除索引） #

解决这个移位问题的关键是从大的索引往小的索引删，这样前面的索引不会被后面的删除操作影响。代码如下：

A = [[0], [0, 1], [0, 1, 2], [0, 1], [0, 1, 2, 3]]
W = [[2, 2], [1, 2, 3], [2, 2, 2, 3], [1, 3, 4, 4], [1, 1, 3, 4]]

for t in range(len(A)):
    # 对要删除的索引进行降序排序
    for idx in sorted(A[t], reverse=True):
        del W[t][idx]

print(W)
# 输出结果：[[2], [3], [3], [4, 4], []]

解决方案2：创建新列表（不修改原数据） #

如果你不想改动原来的W列表，更直观的方式是用列表推导式筛选出不需要删除的元素，这种方法不需要考虑索引顺序，逻辑更清晰：

A = [[0], [0, 1], [0, 1, 2], [0, 1], [0, 1, 2, 3]]
W = [[2, 2], [1, 2, 3], [2, 2, 2, 3], [1, 3, 4, 4], [1, 1, 3, 4]]

new_W = [
    # 保留那些索引不在A[t]里的元素
    [elem for i, elem in enumerate(w_sub) if i not in set(a_sub)]
    for w_sub, a_sub in zip(W, A)
]

print(new_W)
# 输出结果和上面一致：[[2], [3], [3], [4, 4], []]

这里把a_sub转成集合是为了让i not in ...的判断更快，尤其是当A的子列表很长的时候。

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