删除列表中前 N 个元素的最有效方法?
Posted
技术标签:
【中文标题】删除列表中前 N 个元素的最有效方法?【英文标题】:The most efficient way to remove first N elements in a list? 【发布时间】:2016-02-11 03:09:45 【问题描述】:我需要从 Python 2.7 中的对象列表中删除前 n 个元素。有没有不使用循环的简单方法?
【问题讨论】:
x.pop(5)
、del x[5]
或 x.remove("cow")
。前两个按索引删除,后一个按搜索条件删除。第二个可以通过 del x[2:5]
删除索引 2-5
为什么是 5?如果我有例如 [a,b,c,d,e,f,g...z] 并且只想保留 [f,g...z]?
那么你想做pos = x.index("f"); end = x.index("z"); del x[pos:end]
你必须考虑到z
可能在f
之前存在,并在搜索@ 时使用pos
作为起点987654329@ 等,但你明白了。
也许你想要列表[n:]
【参考方案1】:
您可以使用列表切片来归档您的目标。
删除前 5 个元素:
n = 5
mylist = [1,2,3,4,5,6,7,8,9]
newlist = mylist[n:]
print newlist
输出:
[6, 7, 8, 9]
或者del
如果你只想使用一个列表:
n = 5
mylist = [1,2,3,4,5,6,7,8,9]
del mylist[:n]
print mylist
输出:
[6, 7, 8, 9]
【讨论】:
【参考方案2】:Python 列表没有在列表的开头进行操作,并且在此操作中非常无效。
虽然你可以写
mylist = [1, 2 ,3 ,4]
mylist.pop(0)
它非常效率低下。
如果您只想从列表中删除项目,可以使用del
执行此操作:
del mylist[:n]
这也非常快:
In [34]: %%timeit
help=range(10000)
while help:
del help[:1000]
....:
10000 loops, best of 3: 161 µs per loop
如果你需要从列表的开头获取元素,你应该使用 Raymond Hettinger 的collections.deque
及其popleft()
方法。
from collections import deque
deque(['f', 'g', 'h', 'i', 'j'])
>>> d.pop() # return and remove the rightmost item
'j'
>>> d.popleft() # return and remove the leftmost item
'f'
比较:
列表 + 弹出(0)
In [30]: %%timeit
....: help=range(10000)
....: while help:
....: help.pop(0)
....:
100 loops, best of 3: 17.9 ms per loop
双端队列 + popleft()
In [33]: %%timeit
help=deque(range(10000))
while help:
help.popleft()
....:
1000 loops, best of 3: 812 µs per loop
【讨论】:
使用 del mylist[:n] 还是低效? 我很好奇:如果list
支持数组,pop
如何高效? wiki.python.org/moin/TimeComplexity
@RedVelvet:是的。尽管如此,关于deque
与list
请记住,双端队列并不具备list
拥有的所有功能。在更大的班次(100 万行)上,我发现 del mylist[1000000:]
与所有行的 deque.popleft()
执行大致相同。所以这取决于您的用例。
哇,实际上在规范中,您可以在调用 popleft()
时迭代双端队列吗?【参考方案3】:
l = [1, 2, 3, 4, 5]
del l[0:3] # Here 3 specifies the number of items to be deleted.
如果你想从列表中删除一些项目,这是代码。您不妨跳过冒号前的零。它没有那么重要。这也可以。
l = [1, 2, 3, 4, 5]
del l[:3] # Here 3 specifies the number of items to be deleted.
【讨论】:
【参考方案4】:尝试运行这段代码:
del x[:N]
【讨论】:
【参考方案5】:l = [5,1,4,2,3,6]
将列表从小到大排序
l.sort()
删除列表中的前 2 项
for _ in range(2)
l.remove(l[0])
打印列表
print(l)
【讨论】:
【参考方案6】:假设你有这个列表:
mylist = [1,2,3,4,5,6,7,8,9]
并且您想删除 x
的最后一个元素并将它们存储在另一个列表中
newlist = [mylist.pop() for _ in range(x)]
您可以修改传递给 pop 的参数,以便从头开始删除元素
newlist = [mylist.pop(0) for _ in range(x)]
或者保留第一个元素,之后删除x
元素
newlist = [mylist.pop(1) for _ in range(x)]
【讨论】:
以上是关于删除列表中前 N 个元素的最有效方法?的主要内容,如果未能解决你的问题,请参考以下文章
Python / numpy:对数组的n个元素求和的最有效方法,以便每个输出元素是前n个输入元素的总和?