Python 跳过前几行读取文件内容 itertools(islice)使用

Python 读取文件时,经常需要跳过第一行读取文件内容。比较容易想到是为每行设置一个 line_num ,然后判断 line_num 是否为1,如果不等于 1 ,则进行读取操作。

相应的 Python 代码如下:

input_file = open("test.csv")  
line_num = 0  
for line in islice(input_file, 1, None):  
    line_num += 1  
    if (line_num != 1):  
        do_readline()

但这样写的代码执行效率偏低,因为每次都需要判断当前的行号是否为 1 。使用 Python 提供的 itertools 工具,我们可以避免此类问题。itertools 的目的就是为了提高 looping 的效率。

相应的代码如下:

from itertools import islice  
input_file = open("test.csv")  
for line in islice(input_file, 1, None):  
    do_readline()

参数:islice(文件名, 开始行, 结束行):

via:https://blog.csdn.net/huangzhiyuan111/article/details/52276749

猜你喜欢

转载自blog.csdn.net/weixin_41287692/article/details/86503774