python - 在 Python 中高效地删除列表中的列表

Question

假设您有一个列表，如 l=[[1,2,3],[1,3,4],[1,5,2],[4,2,1]]，并且您想删除 l 中索引 1 处有 2 个的每个列表。最有效的方法是什么？目前我使用 while 循环遍历每个项目并单独检查，但是当列表变长时这会占用大量 CPU。有没有一种方法可以使用 .remove() 函数来简化此过程？

Answer 1

您可以使用列表理解 来过滤您给定的列表。列表理解是一种将一个列表(实际上是任何 iterable)转换为另一个列表的优雅方式。

list_updated = [item for item in l if item[1] != 2]

相当于。

result = []
for item in l:
  if item[1] != 2:
    result.append(item)

对于大列表，您可以使用生成器理解。 生成器 理解是列表理解的惰性 版本。

list_updated = (item for item in l if item[1] != 2)

使用生成器，您可以使用next 方法查看结果。