Python中如何基于行索引删除列表元素且不打乱索引？

假设我有一个元素列表，想要基于一组行索引删除其中部分元素。示例如下：

l = ['ciao','hopefully','we','are','going','to','sort','this','out']
idx = [1,3,5]

若执行以下代码会失效，因为循环未考虑删除元素后列表长度变化导致索引偏移：

for x in idx:
    del l[x]

# 问题：仅能正确删除idx的第一个元素，后续因列表缩短，原索引不再对应更新后的列表

注：由于该列表来自网页抓取，无法转为数组使用np.delete方法。求问如何一次性基于行索引删除列表元素且不打乱索引？

解决方案 #

方法1：按索引从大到小删除 #

核心思路是先删大索引，再删小索引——大索引的元素被删除后，不会影响更小索引的元素位置，自然不会出现偏移问题。只需把索引列表降序排序后遍历删除即可：

l = ['ciao','hopefully','we','are','going','to','sort','this','out']
idx = [1,3,5]

# 对索引列表降序排序后执行删除
for x in sorted(idx, reverse=True):
    del l[x]

print(l)  # 输出结果：['ciao', 'we', 'going', 'sort', 'this', 'out']

方法2：生成新列表过滤指定索引 #

如果不需要修改原列表，直接生成一个新列表，只保留不在目标索引里的元素即可。用集合存储索引能大幅提升查找效率：

l = ['ciao','hopefully','we','are','going','to','sort','this','out']
idx = [1,3,5]

# 把索引转成集合，加快查找速度
idx_set = set(idx)
new_l = [item for i, item in enumerate(l) if i not in idx_set]

print(new_l)  # 输出结果：['ciao', 'we', 'going', 'sort', 'this', 'out']

两种方法对比 #

• 方法1：直接修改原列表，无需额外内存，适合需要原地修改的场景。
• 方法2：生成新列表，不会破坏原数据，更安全，适合需要保留原列表的场景，集合查找的效率比列表更高。

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