Python 标准库 fileinput与文件迭代器

目录

• 1. 使用 fileinput 进行迭代
• 2. 文件迭代器
• 3. 字典的 setdefault 方法使用举例

1. 使用 fileinput 进行迭代

fileinput 模块可以对一个或多个文件中的内容进行迭代、遍历等操作。该模块的 input() 函数有点类似文件 readlines 函数，但区别明显：

• readlines 是一次性读取文件中的全部内容，如果文件较大的话，会占用大量内存；
• input 返回的则是一个迭代对象，结合 for 循环使用，典型用法如下：

import fileinput

for line in fileinput.input(filename):
    process(line)

实例：

import fileinput

for line in fileinput.input("D:/Users/data.txt"):
    print(line)

注意：input 函数有多个参数：分别设置读取文件路径、读写模式、编码方式、缓冲区大小、备份文件扩展名等。

此外，fileinput 模块还有很多常用的函数，这里介绍几个最常用的。

函数名 说明：

• fileinput.input() 返回能够用于 for 循环遍历的对象
• fileinput.filename() 返回当前文件的名称
• fileinput.lineno() 返回当前已经读取的行的数量（或者序号）
• fileinput.filelineno() 回当前读取的行的行号
• fileinput.isfirstline() 检查当前行是否是文件的第一行
• fileinput.isstdin() 判断最后一行是否从 stdin 中读取
• fileinput.close() 关闭队列

2. 文件迭代器

#打开文件，一次写入多行内容
f = open("D:/Users/data.txt",'w')
content = ['append information1\n','append information2']
f.writelines(content)
f.close()

#打开文件，通过文件迭代器遍历文件
f = open("D:/Users/data.txt",'r')
for line in f:
    print('content:',line)
# 或者 将读取内容转化为字符串列表 效果类似 readlines
lines = list(f)
print(lines)
#关闭文件
f.close()

执行结果（省略空行）：

content: append information1
content: append information2

3. 字典的 setdefault 方法使用举例

如果仅当字典中不存在某个键值对时，才插入到字典中；如果存在，不必插入(也就不会修改键值对)。

这种场景，使用字典自带方法 setdefault :

In [35]: d = {'a':1,'b':2}

In [36]: r = d.setdefault('c',3) # r: 3

In [37]: r
Out[37]: 3

In [38]: d
Out[38]: {'a': 1, 'b': 2, 'c': 3}


In [39]: r = d.setdefault('c',33) # r:3，已经存在'c':3的键值对，所以setdefault时d无改变

In [40]: r
Out[40]: 3

In [41]: d
Out[41]: {'a': 1, 'b': 2, 'c': 3}

到此这篇关于Python 标准库 fileinput与文件迭代器的文章就介绍到这了,更多相关Python fileinput 内容请搜索自由互联以前的文章或继续浏览下面的相关文章希望大家以后多多支持自由互联！