fileinput 的用法及功能
使用 open 函数去读取文件，似乎是所有 python 工程师的共识。
今天明哥要给大家推荐一个比 open 更好用、更优雅的读取文件方法 -- 使用 fileinput
fileinput 是 python 的内置模块，但我相信，不少人对它都是陌生的。今天我把 fileinput 的所有的用法、功能进行详细的讲解，并列举了一些非常实用的案例，对于理解和使用它可以说完全没有问题。
1. 从标准输入中读取当你的 python 脚本没有传入任何参数时，fileinput 默认会以 stdin 作为输入源
# demo.pyimport fileinputfor line in fileinput.input(): print(line)效果如下，不管你输入什么，程序会自动读取并再打印一次，像个复读机似的。
$ python demo.py hellohellopythonpython2. 单独打开一个文件单独打开一个文件，只需要在 files 中输入一个文件名即可
import fileinputwith fileinput.input(files=('a.txt',)) as file: for line in file: print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='')其中 a.txt 的内容如下
helloworld执行后就会输出如下
$ python demo.pya.txt 第1行: helloa.txt 第2行: world需要说明的一点是，fileinput.input() 默认使用 mode='r' 的模式读取文件，如果你的文件是二进制的，可以使用mode='rb' 模式。fileinput 有且仅有这两种读取模式。
3. 批量打开多个文件从上面的例子也可以看到，我在 fileinput.input 函数中传入了 files 参数，它接收一个包含多个文件名的列表或元组，传入一个就是读取一个文件，传入多件就是读取多个文件。
import fileinputwith fileinput.input(files=('a.txt', 'b.txt')) as file: for line in file: print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='')a.txt 和 b.txt 的内容分别是
$ cat a.txthelloworld$ cat b.txthellopython运行后输出结果如下，由于 a.txt 和 b.txt 的内容被整合成一个文件对象 file ，因此 fileinput.lineno() 只有在读取一个文件时，才是原文件中真实的行号。
$ python demo.pya.txt 第1行: helloa.txt 第2行: worldb.txt 第3行: hellob.txt 第4行: python如果想要在读取多个文件的时候，也能读取原文件的真实行号，可以使用 fileinput.filelineno() 方法
import fileinputwith fileinput.input(files=('a.txt', 'b.txt')) as file: for line in file: print(f'{fileinput.filename()} 第{fileinput.filelineno()}行: {line}', end='')运行后，输出如下
$ python demo.pya.txt 第1行: helloa.txt 第2行: worldb.txt 第1行: hellob.txt 第2行: python这个用法和 glob 模块简直是绝配
import fileinputimport globfor line in fileinput.input(glob.glob(*.txt)): if fileinput.isfirstline(): print('-'*20, f'reading {fileinput.filename()}...', '-'*20) print(str(fileinput.lineno()) + ': ' + line.upper(), end=)运行效果如下
$ python demo.py-------------------- reading b.txt... --------------------1: hello2: python-------------------- reading a.txt... --------------------3: hello4: world4. 读取的同时备份文件fileinput.input 有一个 backup 参数，你可以指定备份的后缀名，比如 .bak
import fileinputwith fileinput.input(files=(a.txt,), backup=.bak) as file: for line in file: print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='')运行的结果如下，会多出一个 a.txt.bak 文件
$ ls -l a.txt*-rw-r--r-- 1 ming staff 12 2 27 10:43 a.txt$ python demo.pya.txt 第1行: helloa.txt 第2行: world$ ls -l a.txt*-rw-r--r-- 1 ming staff 12 2 27 10:43 a.txt-rw-r--r-- 1 ming staff 42 2 27 10:39 a.txt.bak5. 标准输出重定向替换fileinput.input 有一个 inplace 参数，表示是否将标准输出的结果写回文件，默认不取代
请看如下一段测试代码
import fileinputwith fileinput.input(files=(a.txt,), inplace=true) as file: print([info] task is started...) for line in file: print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='') print([info] task is closed...)运行后，会发现在 for 循环体内的 print 内容会写回到原文件中了。而在 for 循环体外的 print 则没有变化。
$ cat a.txthelloworld$ python demo.py[info] task is started...[info] task is closed...$ cat a.txt a.txt 第1行: helloa.txt 第2行: world利用这个机制，可以很容易的实现文本替换。
import sysimport fileinputfor line in fileinput.input(files=('a.txt', ), inplace=true): #将windows/dos格式下的文本文件转为linux的文件 if line[-2:] == rn: line = line + n sys.stdout.write(line)附：如何实现 dos 和 unix 格式互换以供程序测试，使用 vim 输入如下指令即可
dos转unix：:setfileformat=unixunix转dos：:setfileformat=dos6. 不得不介绍的方法如果只是想要 fileinput 当做是替代 open 读取文件的工具，那么以上的内容足以满足你的要求。
fileinput.filenam()
返回当前被读取的文件名。在第一行被读取之前，返回 none。fileinput.fileno()
返回以整数表示的当前文件“文件描述符”。当未打开文件时（处在第一行和文件之间），返回 -1。fileinput.lineno()
返回已被读取的累计行号。在第一行被读取之前，返回 0。在最后一个文件的最后一行被读取之后，返回该行的行号。fileinput.filelineno()
返回当前文件中的行号。在第一行被读取之前，返回 0。在最后一个文件的最后一行被读取之后，返回此文件中该行的行号。但若要想基于 fileinput 来做一些更加复杂的逻辑，也许你会需要用到如下这几个方法
fileinput.isfirstline()
如果刚读取的行是其所在文件的第一行则返回 true，否则返回 false。fileinput.isstdin()
如果最后读取的行来自 sys.stdin 则返回 true，否则返回 false。fileinput.nextfile()
关闭当前文件以使下次迭代将从下一个文件（如果存在）读取第一行；不是从该文件读取的行将不会被计入累计行数。直到下一个文件的第一行被读取之后文件名才会改变。在第一行被读取之前，此函数将不会生效；它不能被用来跳过第一个文件。在最后一个文件的最后一行被读取之后，此函数将不再生效。fileinput.close()
关闭序列。7. 进阶一点的玩法在 fileinput.input() 中有一个 openhook 的参数，它支持用户传入自定义的对象读取方法。
若你没有传入任何的勾子，fileinput 默认使用的是 open 函数。
fileinput 为我们内置了两种勾子供你使用
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))如果你自己的场景比较特殊，以上的三种勾子都不能满足你的要求，你也可以自定义。
这边我举个例子来抛砖引玉下
假如我想要使用 fileinput 来读取网络上的文件，可以这样定义勾子。
先使用 requests 下载文件到本地再使用 open 去读取它def online_open(url, mode): import requests r = requests.get(url) filename = url.split(/)[-1] with open(filename,'w') as f1: f1.write(r.content.decode(utf-8)) f2 = open(filename,'r') return f2直接将这个函数传给 openhook 即可
import fileinputfile_url = 'https://www.csdn.net/robots.txt'with fileinput.input(files=(file_url,), openhook=online_open) as file: for line in file: print(line, end=)运行后按预期一样将 csdn 的 robots 的文件打印了出来
user-agent: * disallow: /scripts disallow: /public disallow: /css/ disallow: /images/ disallow: /content/ disallow: /ui/ disallow: /js/ disallow: /scripts/ disallow: /article_preview.html* disallow: /tag/disallow: /*?*disallow: /link/sitemap: https://www.csdn.net/sitemap-aggpage-index.xmlsitemap: https://www.csdn.net/article/sitemap.txt8. 列举一些实用案例案例一 ：读取一个文件所有行
import fileinputfor line in fileinput.input('data.txt'): print(line, end=)案例二 ：读取多个文件所有行
import fileinputimport globfor line in fileinput.input(glob.glob(*.txt)): if fileinput.isfirstline(): print('-'*20, f'reading {fileinput.filename()}...', '-'*20) print(str(fileinput.lineno()) + ': ' + line.upper(), end=)案例三 ：利用fileinput将crlf文件转为lf
import sysimport fileinputfor line in fileinput.input(files=('a.txt', ), inplace=true): #将windows/dos格式下的文本文件转为linux的文件 if line[-2:] == rn: line = line + n sys.stdout.write(line)案例四 ：配合 re 做日志分析：取所有含日期的行
#--样本文件--：error.logaaa1970-01-01 13:45:30 error: **** due to system disk spacke not enough...bbb1970-01-02 10:20:30 error: **** due to system out of memory...ccc#---测试脚本---import reimport fileinputimport syspattern = 'd{4}-d{2}-d{2} d{2}:d{2}:d{2}'for line in fileinput.input('error.log',backup='.bak',inplace=1): if re.search(pattern,line): sys.stdout.write(= > ) sys.stdout.write(line)#---测试结果---= > 1970-01-01 13:45:30 error: **** due to system disk spacke not enough...= > 1970-01-02 10:20:30 error: **** due to system out of memory...案例五 ：利用fileinput实现类似于grep的功能
import sysimport reimport fileinputpattern= re.compile(sys.argv[1])for line in fileinput.input(sys.argv[2]): if pattern.match(line): print(fileinput.filename(), fileinput.filelineno(), line)$ ./demo.py import.*re *.py#查找所有py文件中，含import re字样的addressbook.py 2 import readdressbook1.py 10 import readdressbook2.py 18 import retest.py 238 import re9. 写在最后fileinput 是对 open 函数的再次封装，在仅需读取数据的场景中， fileinput 显然比 open 做得更专业、更人性，当然在其他有写操作的复杂场景中，fileinput 就无能为力啦，本身从 fileinput 的命名上就知道这个模块只专注于输入（读）而不是输出（写）。
农残检测仪的性能介绍
基于精密仪表放大器的惠斯登电桥解决方案
专业解读锂电池的正极材料
高通骁龙™855移动平台,性能与功耗的巧妙平衡
TI车规级处理器平台创新不止，持续赋能自动驾驶技术
fileinput 的用法及功能
随着新能源补贴5.5亿元到账 安凯汽车今年将扭亏为盈
关于VillaKit语音控制研讨会的内容分析和介绍
什么是滑动均值滤波？matlab如何实现滑动平均滤波？详细分析
创想三维：3D扫描仪在3D打印机上的运用
新一代多光谱偏振光生理成像技术研发成功
高通835：性能提升功耗降低，首发机型除了S8还有小米6
如何引领AI行业健康平稳发展
隆基与巴基斯坦经销商签署的Hi-MO X6框架协议达到1280MW
直流快速充电器启动方案
三端固定式集成稳压器的产品分类
有机硅导热胶，可同时满足电子元件固定和导热的两大需求
立即预约参与 | 研扬智能边缘计算线上研讨会-华为昇腾专场
美高森美选择英特尔代工服务开发数字集成电路
彭于晏代言！vivo大屏自拍强机X9 Plus来了：3498元
今天明哥要给大家推荐一个比 open 更好用、更优雅的读取文件方法 -- 使用 fileinput
fileinput 是 python 的内置模块，但我相信，不少人对它都是陌生的。今天我把 fileinput 的所有的用法、功能进行详细的讲解，并列举了一些非常实用的案例，对于理解和使用它可以说完全没有问题。
1. 从标准输入中读取当你的 python 脚本没有传入任何参数时，fileinput 默认会以 stdin 作为输入源
# demo.pyimport fileinputfor line in fileinput.input(): print(line)效果如下，不管你输入什么，程序会自动读取并再打印一次，像个复读机似的。
$ python demo.py hellohellopythonpython2. 单独打开一个文件单独打开一个文件，只需要在 files 中输入一个文件名即可
import fileinputwith fileinput.input(files=('a.txt',)) as file: for line in file: print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='')其中 a.txt 的内容如下
helloworld执行后就会输出如下
$ python demo.pya.txt 第1行: helloa.txt 第2行: world需要说明的一点是，fileinput.input() 默认使用 mode='r' 的模式读取文件，如果你的文件是二进制的，可以使用mode='rb' 模式。fileinput 有且仅有这两种读取模式。
3. 批量打开多个文件从上面的例子也可以看到，我在 fileinput.input 函数中传入了 files 参数，它接收一个包含多个文件名的列表或元组，传入一个就是读取一个文件，传入多件就是读取多个文件。
import fileinputwith fileinput.input(files=('a.txt', 'b.txt')) as file: for line in file: print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='')a.txt 和 b.txt 的内容分别是
$ cat a.txthelloworld$ cat b.txthellopython运行后输出结果如下，由于 a.txt 和 b.txt 的内容被整合成一个文件对象 file ，因此 fileinput.lineno() 只有在读取一个文件时，才是原文件中真实的行号。
$ python demo.pya.txt 第1行: helloa.txt 第2行: worldb.txt 第3行: hellob.txt 第4行: python如果想要在读取多个文件的时候，也能读取原文件的真实行号，可以使用 fileinput.filelineno() 方法
import fileinputwith fileinput.input(files=('a.txt', 'b.txt')) as file: for line in file: print(f'{fileinput.filename()} 第{fileinput.filelineno()}行: {line}', end='')运行后，输出如下
$ python demo.pya.txt 第1行: helloa.txt 第2行: worldb.txt 第1行: hellob.txt 第2行: python这个用法和 glob 模块简直是绝配
import fileinputimport globfor line in fileinput.input(glob.glob(*.txt)): if fileinput.isfirstline(): print('-'*20, f'reading {fileinput.filename()}...', '-'*20) print(str(fileinput.lineno()) + ': ' + line.upper(), end=)运行效果如下
$ python demo.py-------------------- reading b.txt... --------------------1: hello2: python-------------------- reading a.txt... --------------------3: hello4: world4. 读取的同时备份文件fileinput.input 有一个 backup 参数，你可以指定备份的后缀名，比如 .bak
import fileinputwith fileinput.input(files=(a.txt,), backup=.bak) as file: for line in file: print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='')运行的结果如下，会多出一个 a.txt.bak 文件
$ ls -l a.txt*-rw-r--r-- 1 ming staff 12 2 27 10:43 a.txt$ python demo.pya.txt 第1行: helloa.txt 第2行: world$ ls -l a.txt*-rw-r--r-- 1 ming staff 12 2 27 10:43 a.txt-rw-r--r-- 1 ming staff 42 2 27 10:39 a.txt.bak5. 标准输出重定向替换fileinput.input 有一个 inplace 参数，表示是否将标准输出的结果写回文件，默认不取代
请看如下一段测试代码
import fileinputwith fileinput.input(files=(a.txt,), inplace=true) as file: print([info] task is started...) for line in file: print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='') print([info] task is closed...)运行后，会发现在 for 循环体内的 print 内容会写回到原文件中了。而在 for 循环体外的 print 则没有变化。
$ cat a.txthelloworld$ python demo.py[info] task is started...[info] task is closed...$ cat a.txt a.txt 第1行: helloa.txt 第2行: world利用这个机制，可以很容易的实现文本替换。
import sysimport fileinputfor line in fileinput.input(files=('a.txt', ), inplace=true): #将windows/dos格式下的文本文件转为linux的文件 if line[-2:] == rn: line = line + n sys.stdout.write(line)附：如何实现 dos 和 unix 格式互换以供程序测试，使用 vim 输入如下指令即可
dos转unix：:setfileformat=unixunix转dos：:setfileformat=dos6. 不得不介绍的方法如果只是想要 fileinput 当做是替代 open 读取文件的工具，那么以上的内容足以满足你的要求。
fileinput.filenam()
返回当前被读取的文件名。在第一行被读取之前，返回 none。fileinput.fileno()
返回以整数表示的当前文件“文件描述符”。当未打开文件时（处在第一行和文件之间），返回 -1。fileinput.lineno()
返回已被读取的累计行号。在第一行被读取之前，返回 0。在最后一个文件的最后一行被读取之后，返回该行的行号。fileinput.filelineno()
返回当前文件中的行号。在第一行被读取之前，返回 0。在最后一个文件的最后一行被读取之后，返回此文件中该行的行号。但若要想基于 fileinput 来做一些更加复杂的逻辑，也许你会需要用到如下这几个方法
fileinput.isfirstline()
如果刚读取的行是其所在文件的第一行则返回 true，否则返回 false。fileinput.isstdin()
如果最后读取的行来自 sys.stdin 则返回 true，否则返回 false。fileinput.nextfile()
关闭当前文件以使下次迭代将从下一个文件（如果存在）读取第一行；不是从该文件读取的行将不会被计入累计行数。直到下一个文件的第一行被读取之后文件名才会改变。在第一行被读取之前，此函数将不会生效；它不能被用来跳过第一个文件。在最后一个文件的最后一行被读取之后，此函数将不再生效。fileinput.close()
关闭序列。7. 进阶一点的玩法在 fileinput.input() 中有一个 openhook 的参数，它支持用户传入自定义的对象读取方法。
若你没有传入任何的勾子，fileinput 默认使用的是 open 函数。
fileinput 为我们内置了两种勾子供你使用
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))如果你自己的场景比较特殊，以上的三种勾子都不能满足你的要求，你也可以自定义。
这边我举个例子来抛砖引玉下
假如我想要使用 fileinput 来读取网络上的文件，可以这样定义勾子。
先使用 requests 下载文件到本地再使用 open 去读取它def online_open(url, mode): import requests r = requests.get(url) filename = url.split(/)[-1] with open(filename,'w') as f1: f1.write(r.content.decode(utf-8)) f2 = open(filename,'r') return f2直接将这个函数传给 openhook 即可
import fileinputfile_url = 'https://www.csdn.net/robots.txt'with fileinput.input(files=(file_url,), openhook=online_open) as file: for line in file: print(line, end=)运行后按预期一样将 csdn 的 robots 的文件打印了出来
user-agent: * disallow: /scripts disallow: /public disallow: /css/ disallow: /images/ disallow: /content/ disallow: /ui/ disallow: /js/ disallow: /scripts/ disallow: /article_preview.html* disallow: /tag/disallow: /*?*disallow: /link/sitemap: https://www.csdn.net/sitemap-aggpage-index.xmlsitemap: https://www.csdn.net/article/sitemap.txt8. 列举一些实用案例案例一 ：读取一个文件所有行
import fileinputfor line in fileinput.input('data.txt'): print(line, end=)案例二 ：读取多个文件所有行
import fileinputimport globfor line in fileinput.input(glob.glob(*.txt)): if fileinput.isfirstline(): print('-'*20, f'reading {fileinput.filename()}...', '-'*20) print(str(fileinput.lineno()) + ': ' + line.upper(), end=)案例三 ：利用fileinput将crlf文件转为lf
import sysimport fileinputfor line in fileinput.input(files=('a.txt', ), inplace=true): #将windows/dos格式下的文本文件转为linux的文件 if line[-2:] == rn: line = line + n sys.stdout.write(line)案例四 ：配合 re 做日志分析：取所有含日期的行
#--样本文件--：error.logaaa1970-01-01 13:45:30 error: **** due to system disk spacke not enough...bbb1970-01-02 10:20:30 error: **** due to system out of memory...ccc#---测试脚本---import reimport fileinputimport syspattern = 'd{4}-d{2}-d{2} d{2}:d{2}:d{2}'for line in fileinput.input('error.log',backup='.bak',inplace=1): if re.search(pattern,line): sys.stdout.write(= > ) sys.stdout.write(line)#---测试结果---= > 1970-01-01 13:45:30 error: **** due to system disk spacke not enough...= > 1970-01-02 10:20:30 error: **** due to system out of memory...案例五 ：利用fileinput实现类似于grep的功能
import sysimport reimport fileinputpattern= re.compile(sys.argv[1])for line in fileinput.input(sys.argv[2]): if pattern.match(line): print(fileinput.filename(), fileinput.filelineno(), line)$ ./demo.py import.*re *.py#查找所有py文件中，含import re字样的addressbook.py 2 import readdressbook1.py 10 import readdressbook2.py 18 import retest.py 238 import re9. 写在最后fileinput 是对 open 函数的再次封装，在仅需读取数据的场景中， fileinput 显然比 open 做得更专业、更人性，当然在其他有写操作的复杂场景中，fileinput 就无能为力啦，本身从 fileinput 的命名上就知道这个模块只专注于输入（读）而不是输出（写）。
农残检测仪的性能介绍
基于精密仪表放大器的惠斯登电桥解决方案
专业解读锂电池的正极材料
高通骁龙™855移动平台,性能与功耗的巧妙平衡
TI车规级处理器平台创新不止，持续赋能自动驾驶技术
fileinput 的用法及功能
随着新能源补贴5.5亿元到账 安凯汽车今年将扭亏为盈
关于VillaKit语音控制研讨会的内容分析和介绍
什么是滑动均值滤波？matlab如何实现滑动平均滤波？详细分析
创想三维：3D扫描仪在3D打印机上的运用
新一代多光谱偏振光生理成像技术研发成功
高通835：性能提升功耗降低，首发机型除了S8还有小米6
如何引领AI行业健康平稳发展
隆基与巴基斯坦经销商签署的Hi-MO X6框架协议达到1280MW
直流快速充电器启动方案
三端固定式集成稳压器的产品分类
有机硅导热胶，可同时满足电子元件固定和导热的两大需求
立即预约参与 | 研扬智能边缘计算线上研讨会-华为昇腾专场
美高森美选择英特尔代工服务开发数字集成电路
彭于晏代言！vivo大屏自拍强机X9 Plus来了：3498元