百木园-与人分享,
就是让自己快乐。

这个Python读取文件的方法,堪称天花板级别...

1、方法介绍
基本用法

先来看一下fileinput的基本功能:

fileinput.filename():返回当前被读取的文件名。
—>在第一行被读取之前,返回 None。

fileinput.fileno():返回以整数表示的当前文件“文件描述符”。
—>当未打开文件时(处在第一行和文件之间),返回 -1。

fileinput.lineno():返回已被读取的累计行号。
—>在第一行被读取之前,返回 0。在最后一个文件的最后一行被读取之后,返回该行的行号。

fileinput.filelineno():返回当前文件中的行号。
—>在第一行被读取之前,返回 0。
—>在最后一个文件的最后一行被读取之后,返回此文件中该行的行号。
 

进阶用法

fileinput.isfirstline():如果刚读取的行是其所在文件的第一行则返回 True,否则返回 False。

fileinput.isstdin():如果最后读取的行来自 sys.stdin 则返回 True,否则返回 False。

fileinput.nextfile():关闭当前文件以使下次迭代将从下一个文件(如果存在)读取第一行;不是从该文件读取的行将不会被计入累计行数。直到下一个文件的第一行被读取之后文件名才会改变。
—>在第一行被读取之前,此函数将不会生效;它不能被用来跳过第一个文件。
—>在最后一个文件的最后一行被读取之后,此函数将不再生效。

fileinput.close():关闭序列。

2、 默认读取

 

代码示例


import fileinput

\'当 Python 脚本没有传入任何参数时,fileinput 默认会以 stdin 作为输入源\'
for line in fileinput.input():
    print(f\'{line}\')

运行结果

 

 

 

你输入的内容,程序都会读取并再输出。

俗称:复读机

3、处理一个文件

 

代码示例

import fileinput

\'files 输入打开文件的名称即可\'
with fileinput.input(files=(\'output.txt\',)) as file:
    for line in file:
        print(f\'{fileinput.filename()} 第{fileinput.lineno()}行:{line}\',end=\'\')

运行结果

 

 

 

解析:

  • fileinput 有且仅有这两种读取模式:‘r’,‘rb’;
  • fileinput.input() 默认使用 mode=‘r’ 的模式读取文件,如果你的文件是二进制的,可以使用mode=‘rb’ 模式。

4、处理批量文件

 

多文件序号连续排序

调用方法

  • fileinput.lineno()方法

代码示例

import fileinput

\'files 输入打开文件的名称即可\'
with fileinput.input(files=(\'output.txt\',\'input.txt\')) as file:
    for line in file:
        #fileinput.lineno() 把两个文件的整合陈一个文件对象file,需要排序输出
        print(f\'{fileinput.filename()} 第{fileinput.lineno()}行: {line}\', end=\'\')
        
        # fileinput.filelineno()两个文件单独读取,需要单独排序
        print(f\'{fileinput.filename()} 第{fileinput.filelineno()}行: {line}\', end=\'\')

 

 

运行结果

 

 

 

多文件序号单独排序

调用方法

 
 

  • fileinput.filelineno()方法

代码示例

 

import fileinput

\'files 输入打开文件的名称即可\'
with fileinput.input(files=(\'test1.txt\',\'test2.txt\')) as file:
    for line in file:       
        # fileinput.filelineno()两个文件单独读取,需要单独排序
        print(f\'{fileinput.filename()} 第{fileinput.filelineno()}行: {line}\', end=\'\')

运行结果

 

 

 

与glob配合用法

在颜值的时代,上面的输出样式,已经无法满足我们的需要了,于是乎,我们就想到了glob。

 

代码示例

import fileinput
import glob

#glob 匹配te开头的txt文件
for line in fileinput.input(glob.glob(\"te*.txt\")):
    if fileinput.isfirstline():
        #输出读取文件
        print(\'=\'*10,f\'读取文件{fileinput.filename()}\',\'=\'*10)
        #fileinput.filelineno()方法读取
    print(str(fileinput.filelineno())+ \':\'+line.upper(),end=\'\')

运行结果

 

 

 

就这颜值,哪个小姐姐能不喜欢呢。

 

5、读取与备份

 

调用方法

 

  • fileinput.input 的backup 参数,可以指定备份的后缀名,比如 .bak

代码示例

import fileinput

#触发backup的动作,源文件内容被修改,对源文件进行backup
with fileinput.input(files=(\"test1.txt\",), backup=\".bak\",inplace=1) as file:
    for line in file:
        print(line.rstrip().replace(\'111111\', \'222222\'))
        print(f\'{fileinput.filename()} 第{fileinput.lineno()}行: {line}\', end=\'\')

 

运行结果

 

 

 

6、重定向替换

 

解析

 

  • 上面的例子, 用到了 inplace参数,表示是否将标准输出的结果写回文件,默认不取代。

代码示例:

import fileinput

#触发backup的动作,源文件内容被修改,对源文件进行backup
with fileinput.input(files=(\"test2.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 则没有变化。

7、进阶

 

openhook含义解析

 

  • 在 fileinput.input() 中有一个 openhook 的参数,它支持用户传入自定义的对象读取方法;
  • 如果没有传入任何勾子,fileinput 默认使用的是 open 函数;

方法介绍

 

fileinput 内置了两种勾子

1、fileinput.hook_compressed(filename, mode)

使用 gzip 和 bz2 模块透明地打开 gzip 和 bzip2 压缩的文件(通过扩展名 ‘.gz’ 和 ‘.bz2’ 来识别);
如果文件扩展名不是 ‘.gz’ 或 ‘.bz2’,文件会以正常方式打开(即使用 open() 并且不带任何解压操作);
使用示例: fi = fileinput.FileInput(openhook=fileinput.hook_compressed)
2、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 即可:

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ

import fileinput
file_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=\"\")

代码整合:

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
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

import fileinput
file_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=\"\")

 

运行结果

 

 

 

总结

 

关于fileinput的介绍,也就介绍到这里。

fileinput本身是对 open 函数的再次封装,所以在读取的cc部分,就比open显得更专业,更优雅,这也是仅限于读取的方面。
在写的方面,相对于open,就不是那么的强悍。

归根结底,fileinput还是一个不错的方法。值得你拥有。


来源:https://www.cnblogs.com/liuliumei/p/16759620.html
本站部分图文来源于网络,如有侵权请联系删除。

未经允许不得转载:百木园 » 这个Python读取文件的方法,堪称天花板级别...

相关推荐

  • 暂无文章