对collections.deque使用切片表示法

时间:2011-08-15 11:14:27

标签: python slice deque

如何在不改变项目的情况下,从以下deque高效,优雅,抒情地提取项目3..6:

from collections import deque
q = deque('',maxlen=10)
for i in range(10,20):
    q.append(i)

slice notation似乎不适用于deque ...

6 个答案:

答案 0 :(得分:53)

import itertools
output = list(itertools.islice(q, 3, 7))

例如:

>>> import collections, itertools
>>> q = collections.deque(xrange(10, 20))
>>> q
deque([10, 11, 12, 13, 14, 15, 16, 17, 18, 19])
>>> list(itertools.islice(q, 3, 7))
[13, 14, 15, 16]

这应该比目前发布的其他解决方案更有效。证明?

[me@home]$ SETUP="import itertools,collections; q=collections.deque(xrange(1000000))"

[me@home]$ python -m timeit  "$SETUP" "list(itertools.islice(q, 10000, 20000))"
10 loops, best of 3: 68 msec per loop

[me@home]$ python -m timeit "$SETUP" "[q[i] for i in  xrange(10000, 20000)]"
10 loops, best of 3: 98.4 msec per loop

[me@home]$ python -m timeit "$SETUP" "list(q)[10000:20000]"
10 loops, best of 3: 107 msec per loop

答案 1 :(得分:4)

我更喜欢这个,它更短,更容易阅读:

output = list(q)[3:6+1]

答案 2 :(得分:3)

output = [q[i] for i in range(3,6+1)]

答案 3 :(得分:2)

我将此添加为新答案,以提供更好的格式。

为简单起见,Shawn的答案是完美的,但如果你经常需要从dequeue得到一个片段,你可能更喜欢将它子类化并添加__getslice__方法。

from collections import deque
from itertools import islice
class deque_slice(deque):
    def __new__(cls, *args):
        return deque.__new__(cls, *args)
    def __getslice__(self, start, end):
        return list(islice(self, start, end))

这不支持设置一个新切片,但您可以使用相同的概念实现自己的自定义__setslice__方法。

答案 4 :(得分:1)

您可以覆盖__getitem__方法并使用SliceableDeque创建一个islice

您应该考虑某些情况(例如,使用负片对islice不起作用)。

这是我一直在使用的:

import itertools
from collections import deque

class SliceableDeque(deque):
    def __getitem__(self, s):
        try:
            start, stop, step = s.start or 0, s.stop or sys.maxsize, s.step or 1
        except AttributeError:  # not a slice but an int
            return super().__getitem__(s)
        else:
            try:
                return list(itertools.islice(self, start, stop, step))
            except ValueError:  # incase of a negative slice object
                length = len(self)
                start, stop = length + start if start < 0 else start, length + stop if stop < 0 else stop
                return list(itertools.islice(self, start, stop, step))

答案 5 :(得分:1)

这是一个古老的问题,但是对于任何将来的旅行者,Python文档明确建议为此使用rotate

  

rotate()方法提供了一种实现双端队列切片和删除的方法。

https://docs.python.org/2/library/collections.html

实现相对简单:

def slice_deque(d, start, stop, step):
    d.rotate(-start)
    slice = list(itertools.islice(d, 0, stop-start, step))
    d.rotate(start)
    return slice

实际上与直接使用islice相同,除了rotate可以更有效地跳到起点。另一方面,它也临时修改了双端队列,这可能是线程安全问题。