Python中如何高效从二维列表筛选指定元素生成新列表？

嘿，这个问题问得很到位！确实，直接用del在循环里删除元素不仅容易因为索引偏移出问题，效率也未必最优，用创建新列表的方式来筛选保留元素是更稳妥高效的思路。

最优实现思路 #

核心是把待删除索引转换成可快速查询的集合，再通过列表推导式一次性生成新列表——集合的查询是O(1)时间复杂度，列表推导式则是Python中创建新列表的高效方式（底层用C实现，比手动循环append快很多）。

具体代码实现 #

lst = [[5,1,2,3],[2,3,4,5,7],[1,10,9,8,7]]
indexes_to_del = [[0,2],[1,2],[1,3],[2,2]]

# 将待删除索引转为元组集合（列表不可哈希，元组可以），实现O(1)查询
del_index_set = set(tuple(idx) for idx in indexes_to_del)

# 列表推导式生成新列表：遍历每一行+每个元素，只保留不在待删集合中的元素
new_lst = [
    [elem for j, elem in enumerate(sublist) if (i, j) not in del_index_set]
    for i, sublist in enumerate(lst)
]

print(new_lst)  # 输出: [[5, 1, 3], [2, 3, 7], [1, 10, 8, 7]]

为什么这是高效的？ #

1. 集合查询优化：如果直接用[i,j] in indexes_to_del，每次查询都是O(k)（k是待删索引的数量），换成集合后查询时间降到O(1)，当待删索引数量较多时，性能提升非常明显。
2. 列表推导式效率：相比手动写for循环+append，列表推导式的执行逻辑在Python底层用C实现，避免了Python层的循环开销，速度更快。
3. 时间复杂度最优：整个过程只遍历原二维列表一次，总时间复杂度是O(M*N)（M是子列表数量，N是所有子列表的元素总数），这已经是理论上的最优复杂度——毕竟你总得检查每个元素是否需要保留。

另外，如果待删索引列表里有重复项，集合会自动去重，避免重复判断，也是个额外的小优势。

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