python - 从右手边切开

Question

对于列表，可以使用切片 my_list[-10:] 来获取最后(最多)10 个元素

我想对 itertools.islice 做同样的事情。在本例中，我有一个 collections.deque。

是否有比以下方法更有效的方法？

from collections import deque
import itertools
my_deque = deque()
my_list = list()
for i in range(100):
    my_deque.append(i)
    my_list.append(i)

然后我们进行切片:

start = max(0, len(my_deque) - 10)
for i in list(itertools.islice(my_deque, start, None)):
    pass

我的时间:

双端队列: 1000000 次循环，3 中最好的:每次循环 962 ns

列表切片: 10000000 次循环，3 中最好的:每次循环 95.9 ns

Answer 1

正如您所发现的，无法对 collections.deque 进行切片。但它支持可以在这种情况下使用的旋转:

last_n = 10
my_deque.rotate(last_n)
for i in itertools.islice(my_deque, last_n):
    pass

在这种情况下，itertools.islice 的问题在于它需要遍历所有元素，直到到达停止点。这是因为它需要与一般的迭代器一起工作，而不仅仅是像 list 和 deque 这样的随机访问容器。因此，在您的情况下，islice 确实必须遍历 deque 中的所有元素。使用 rotate 然后使用 islice 它只需迭代 10 个以上的元素。

至于时间，很难知道你比较了什么但使用这个设置:

from collections import deque
import itertools
my_deque = deque(range(10000))
my_list = list(range(10000))

我得到以下时间:

%%timeit

my_deque.rotate(10)
for i in itertools.islice(my_deque, 10):
    pass
my_deque.rotate(-10)  # so the next timing operates on the orginal deque again

2.76 µs ± 41.6 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

%%timeit

start = max(0, len(my_deque) - 10)
for i in itertools.islice(my_deque, start, None):
    pass

136 µs ± 8.08 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)

%%timeit

start = max(0, len(my_list) - 10)
for i in itertools.islice(my_list, start, None):
    pass

119 µs ± 1.64 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)

%timeit my_list[-10:]

434 ns ± 12.5 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

因此它无法击败列表切片(仍然慢 ~5 倍)但它肯定比使用你的 islice 方法快得多(~50 倍)。