如何同时从列表中删除多个索引?

时间:2012-07-03 01:06:53

标签: python

说我在这里有这个清单:

list = [a, b, c, d, e, f, g]

如何同时删除说索引2, 3, 45

pop不接受多个值。我怎么做呢?

8 个答案:

答案 0 :(得分:151)

您需要在循环中执行此操作,没有内置操作一次删除多个索引。

您的示例实际上是一个连续的索引序列,因此您可以执行此操作:

del my_list[2:6]

删除从2开始到6之前结束的切片。

从您的问题中不清楚一般情况下是否需要删除任意索引集合,或者它是否始终是连续的序列。

如果你有一个任意的索引集合,那么:

indexes = [2, 3, 5]
for index in sorted(indexes, reverse=True):
    del my_list[index]

请注意,您需要以相反的顺序删除它们,以免丢弃后续索引。

答案 1 :(得分:24)

remove_indices = [1,2,3]
somelist = [i for j, i in enumerate(somelist) if j not in remove_indices]

示例:

In [9]: remove_indices = [1,2,3]

In [10]: somelist = range(10)

In [11]: somelist = [i for j, i in enumerate(somelist) if j not in remove_indices]

In [12]: somelist
Out[12]: [0, 4, 5, 6, 7, 8, 9]

答案 2 :(得分:13)

对于不同的方式,没有多少关于性能的提示,所以我在所有3种通常不同的方法中从50000中删除5000个项目进行了测试,对我来说numpy是赢家(如果你有适合的元素) numpy的):

  • 列举列表理解为7.5秒[另一台PC为4.5秒]
  • 以相反顺序删除项目0.08秒[0.017(!)sec]
  • 0.009秒为numpy.delete [0.006秒]

这里是我定时的代码(如果直接在numpy数组上工作,可以删除第三个函数从/到列表的转换):

import time
import numpy as np
import random

def del_list_indexes(l, id_to_del):
    somelist = [i for j, i in enumerate(l) if j not in id_to_del]
    return somelist

def del_list_inplace(l, id_to_del):
    for i in sorted(id_to_del, reverse=True):
        del(l[i])

def del_list_numpy(l, id_to_del):
    arr = np.array(l, dtype='int32')
    return list(np.delete(arr, id_to_del))

l = range(50000)
random.shuffle(l)
remove_id = random.sample(range(len(l)), 5000) # 10% ==> 5000

# ...

答案 3 :(得分:12)

如果它们是连续的,你可以这样做

x[2:6] = []

如果你想删除不连续的索引,那就有点棘手了。

x = [v for i,v in enumerate(x) if i not in frozenset((2,3,4,5))] 

答案 4 :(得分:7)

如果您可以使用numpy,则可以删除多个索引:

>>> import numpy as np
>>> a = np.arange(10)
>>> np.delete(a,(1,3,5))
array([0, 2, 4, 6, 7, 8, 9])

如果你使用np.r_,你可以将切片与单个索引结合起来:

>>> np.delete(a,(np.r_[0:5,7,9]))
array([5, 6, 8])

但是,删除不是in place,因此您必须为其分配。

答案 5 :(得分:4)

lst = ['a', 'b', 'c', 'd', 'e', 'f', 'g'];
lst = lst[0:2] + lst[6:]

这是一步操作。它不使用循环,因此执行速度很快。它使用列表切片。

答案 6 :(得分:0)

老问题,但我有一个答案。

首先,仔细阅读列表中的元素:

for x in range(len(yourlist)):
    print '%s: %s' % (x, yourlist[x])

然后,使用您要弹出的元素的 索引 列表调用此函数。它足够强大,列表的顺序无关紧要。

def multipop(yourlist, itemstopop):
    result = []
    itemstopop.sort()
    itemstopop = itemstopop[::-1]
    for x in itemstopop:
        result.append(yourlist.pop(x))
    return result

作为奖励,结果应该只包含您想要删除的元素。

  

在[73]中:mylist = ['a','b','c','d','charles']

     

在[76]中:对于范围内的x(len(mylist)):

      mylist[x])
     

...

     

0:a

     

1:b

     

2:c

     

3:d

     

4:查尔斯

     

...

     

在[77]中:multipop(mylist,[0,2,4])

     

Out [77]:['charles','c','a']

     

...

     

在[78]中:mylist

     

Out [78]:['b','d']

答案 7 :(得分:0)

另一种选择(就地,任何指数组合):

_marker = object()

for i in indices:
    my_list[i] = _marker  # marked for deletion

obj[:] = [v for v in my_list if v is not _marker]