python中的文件读取库fileinput
简介
fileinput是open方法的高级封装
- input()函数类似于open的readlines(),区别在于前者是个迭代对象,一次只读一行,后者是一次性读取所有行。
常用函数
fileinput.input() # 进行读取文件对象的初始化,返回能够用于for循环遍历的对象
# return FileInput(files, inplace, backup, mode=mode, openhook=openhook)
fileinput.filename() # 返回当前文件的名称
# return fileInput.filename()
fileinput.lineno() # 返回当前已经读取的行的数量(或者序号)
# return fileInput.lineno()
fileinput.filelineno() # 返回当前读取的行的行号
# return fileInput.filelineno()
fileinput.isfirstline() # 检查当前行是否是文件的第一行
# return fileInput.isfirstline()
fileinput.isstdin() # 判断最后一行是否从stdin中读取
# return fileInput.isstdin()
fileinput.close() # 关闭队列,也是关闭文件
fileinput.nextfile() # 关闭当前文件以使下次迭代将从下一个文件(如果存在)读取第一行;不是从该文件读取的行将不会被计入累计行数。直到下一个文件的第一行被读取之后文件名才会改变。
# —>在第一行被读取之前,此函数将不会生效;它不能被用来跳过第一个文件。
# —>在最后一个文件的最后一行被读取之后,此函数将不再生效。
# 过程中可以使用from glob import glob 方法如(file = glob("*.txt"))
函数参数
打开文件
fileinput.input(files=None, inplace=False, backup="", bufsize=0, *, mode="r", openhook=None)
files:默认是stdin方式 ,传入要打开的文件,可以传入一个文件列表,打开多个文件
inplace:是否将标准输出的结果写回文件,默认不取代
backup:备份文件的扩展名,只指定扩展名,如.bak。如果该文件的备份文件已存在,则会自动覆盖。
bufsize:缓冲区大小,默认为0,如果文件很大,可以修改此参数,一般默认即可
mode: 读写模式,默认为只读
openhook:钩子函数, 该钩子用于控制打开的所有文件,比如说编码方式等,返回一个文件对象
钩子函数
openhook 的参数
- 如果没有传入任何勾子,fileinput 默认使用的是 open 函数
内置函数
fileinput.hook_compressed(filename, mode)
使用 gzip 和 bz2 模块透明地打开 gzip 和 bzip2 压缩的文件(通过扩展名 ‘.gz’ 和 ‘.bz2’ 来识别);
如果文件扩展名不是 ‘.gz’ 或 ‘.bz2’,文件会以正常方式打开(即使用 open() 并且不带任何解压操作);
使用示例:
fi = fileinput.FileInput(openhook=fileinput.hook_compressed)
fileinput.hook_encoded(encoding, errors=None)
返回一个通过 open() 打开每个文件的钩子,使用给定的 encoding 和 errors 来读取文件。
使用示例:
fi = fileinput.FileInput(openhook=fileinput.hook_encoded(“utf-8”, “surrogateescape”))