python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python  fileinput

Python 标准库 fileinput与文件迭代器

作者:wohu1104

这篇文章主要介绍了Python标准库fileinput与文件迭代器,fileinput模块可以对一个或多个文件中的内容进行迭代、遍历等操作,更多详细内容需要的朋友可以参考一下

1. 使用 fileinput 进行迭代

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

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 模块还有很多常用的函数,这里介绍几个最常用的。

函数名 说明:

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 内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文