Python-对迭代器进行切片操作-itertools模块

案例：

对于某个文件，我只想读取到其中100~200行之间的内容，是否可以通过切片的方式进行读取？

我想:

f = open()

f[100:200]

可行？

如何解决这个问题？

方法1：

　　　　全部读取到内存中来

f = open()

f_all = f.readlines()

for i in f_all[100:200]:

       print(i)

# 出现问题，假如文件有20多G，那内存不够

　　方法2：

　　　　把这个文件变成可迭代对象，通过 itertools 中 islice 方法

f = open()

f_need = islice(f, 100, 200)

for i in f_need:

    print(i)

解决该问题的思路

#!/usr/bin/python3

from itertools import islice

# 生成 1~20 的可迭代对象

l = range(20)

# ~5，不包括5

for i in islice(l, 5):

    print(i)

print("_"*30)

# 5~10，不包括10

for i in islice(l, 5, 10):

    print(i)

print("_"*30)

# 5~最后

for i in islice(l, 5, None):

    print(i)

print("_"*30)

# t对象，迭代会消耗t对象的生成器

t = iter(l)

for i in islice(t, 5, 10):

    print(i)

print("_"*30)

# 再次对t迭代，会从10开始

for x in t:

    print(x)

# 解释

# islice(可迭代对象, start, end),顾头不顾尾，

# 只取前5行数据 islice(f, 5)

# 只取5行以后的数据islice(f, 5，None)

# 不支持负数引索，每次迭代，消耗掉内部f生成器

# 本质上是从f的起始位置开始迭代，前面不满足条件的抛弃掉了

巴特西

Python-对迭代器进行切片操作-itertools模块

最新文章

热门文章