Python如何根据另一个索引列表移除指定列表的对应元素

原有代码问题 #

你现有的代码存在两处可优化的冗余逻辑：

• 多余的while循环，循环内部直接break，完全可以删除
• 使用values.remove(x)会额外遍历列表查找元素x的位置，你已经明确要删除的索引是position[i]，直接按索引删除即可

优化方案 #

方案1：小数据量快速优化（时间复杂度O(n²)，实际运行速度提升数倍） #

直接改用pop方法按索引删除元素，去掉冗余逻辑，代码如下：

position = [1, 0, 0]
values = [2, 6, 1]
for pos in position:
    removed = values.pop(pos)
    print(removed)
# 最终values即为处理后的空列表

这个方案写法最简单，适合列表长度在数千级别以内的场景。 #

方案2：大数据量高效优化（时间复杂度近似O(n)） #

如果列表长度达到上万甚至十万级别，频繁pop中间元素带来的数组移位开销会非常高，可以用并查集（DSU）离线处理所有删除操作，避免动态修改列表：

class DSU:
    def __init__(self, size):
        # 多申请1位空间防止边界越界
        self.parent = list(range(size + 1))
    
    def find(self, x):
        # 路径压缩优化查询速度
        if self.parent[x] != x:
            self.parent[x] = self.find(self.parent[x])
        return self.parent[x]

position = [1, 0, 0]
values = [2, 6, 1]
n = len(values)
dsu = DSU(n)
removed_list = []

for pos in position:
    # 查找当前第pos个有效元素对应的原数组索引
    origin_idx = dsu.find(pos)
    removed_list.append(values[origin_idx])
    # 标记该索引已删除，指向后续第一个有效索引
    dsu.parent[origin_idx] = dsu.find(origin_idx + 1)

# 输出删除顺序和原逻辑一致
for num in removed_list:
    print(num)
# 如需获取处理后剩余的元素，可通过以下方式生成
remaining_values = [values[i] for i in range(n) if dsu.find(i) == i]
print(remaining_values) # 输出 []

该方案的总时间复杂度为O(n α(n))，其中α是阿克曼函数的反函数，性能接近线性，适合处理大规模数据。 #

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