删除pythonlist中的项目

2023年10月25日 261次阅读

和b.我想删除列表b中存在“0”的所有数字以及与b中的零共享相同索引的相应数字.

这是我的代码：

a = [ 1 , 23 , 3 , 45 , 5 , 63 , 7 , 8 , 9 , 10 , 11 , 12 , 13 , 14 , 15]

b = [ 8 , 0  , 0 , 7  , 0 , 9  , 3 , 2 , 4 , 13 , 25 , 45 , 34 , 25 , 11]


indexzeroes = [i for i, j in enumerate(b) if j == 0]


for i in indexzeroes:
    b.pop(i)
    a.pop(i)

print a
print b

但是我得到了a和b的错误更新列表.我已经确定了原因是在’for循环’中我每次弹出一个项目时都改变了列表结构,这样剩下的零的索引也会发生变化.

对于这样一个表面上简单的问题,这似乎很复杂.有人可以帮忙吗？

最佳答案 原因是因为当您从a或b弹出第i个索引时,所有元素都会向左移动一次.您可以通过反向弹出项目来解决此问题(因为索引应按排序顺序)：

for i in reversed(indexzeroes):
    a.pop(i)
    b.pop(i)

说的是,这对于大型列表来说可能有些低效(最坏情况为O(n ^ 2)).你最好使用一个能够以一点额外内存为代价来提供O(n)算法的集合：

indexzeroes = {i for i, j in enumerate(b) if j == 0}
a = [x for i, x in enumerate(a) if i not in indexzeros]
b = [x for i, x in enumerate(b) if i not in indexzeros]
点赞