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()
会消耗掉所提供的迭代器中的数据,由于迭代器中的元素只能访问一次,所有如果之后还需要去访问,那就应该先将数据转到列表中去。