Python

itertools.islice() --对迭代器做切片操作

2018-10-03  本文已影响10人  cook__

问题:迭代器和生成器是没没法执行普通的切片操作的,这是因为不知道它们的长度是多少(而且它们也没有实现索引),那么如何切片?

解决方案:使用itertools.islice()来对迭代器和生成器做切片操作
def count(n):
    while True:
        yield n
        n += 1

c = count(0)  # 生成器的绝妙之处:它只会在迭代时才会运行,所以死循环也没有问题,返回一个generator
# print(c[:]) # TypeError: 'generator' object is not subscriptable

import itertools
for x in itertools.islice(c, 10, 13):
    print(x)
Outs:
10
11
12
总结:

1、islice()产生的结果是一个迭代器,它可以产生出所需要的切片元素,但这是通过访问并丢弃所有起始索引之前的元素来实现的;
之后的元素会由islice对象产生出来,直到到达结束索引为止;
2、islice()会消耗掉所提供的迭代器中的数据,由于迭代器中的元素只能访问一次,所有如果之后还需要去访问,那就应该先将数据转到列表中去。

上一篇下一篇

猜你喜欢

热点阅读