菜鸟笔记
提升您的技术认知

一日一技:如何使用 python批量读取多个文件-ag真人游戏


file_list = ['1.txt', '2.txt', '3.txt']
for path in file_list:
    with open(path, encoding='utf-8') as f:
        for line in f:
            print(line)

但这样写代码不好看。python 自带一个更好用的模块:fileinput。

如果要使用 fileinput读取列表中的多个文件,那么可以这样写代码:


import fileinput
file_list = ['1.txt', '2.txt', '3.txt']
with fileinput.input(files=file_list) as f:
    for line in f:
        print(line)

但如果仅仅是这样,那这个模块也并没有省多少事情。我们再来看看它的高级功能。

创建一个 read.py,其内容如下:


import fileinput
with fileinput.input() as f:
    for line in f:
        print(line)

这个代码初看起来,没有读入任何文件,那么它的内容从哪里来呢?

此时我们运行,并尝试输入一些内容回车。如下图所示:

可以看出,它会自动把你输入的内容打印出来,相当于在 whiletrue里面加上了 input。

不仅如此,这段代码不做任何修改,我们在 read.py同目录下创建3个文件 1.txt 2.txt 3.txt。然后使用如下命令运行:


python3 read.py 1.txt 2.txt 3.txt

运行效果如下图所示:

自动把参数对应的文件都读入并打印了出来。这里的参数可以有任意多个。

网站地图