Python中的 fileinput.filename()

简介

fileinput 是一个非常方便的模块，它允许我们一次性处理多个文件。当我们需要同时处理多个文件时，fileinput 模块就可以派上用场。而 fileinput.filename() 方法可用于获取当前正在处理的文件名。

语法

fileinput.filename()

参数

该方法没有参数。

返回值

该方法返回当前正在处理的文件名。

示例代码

import fileinput

for line in fileinput.input(files=('file1.txt', 'file2.txt')):
    if fileinput.filename() == 'file1.txt':
        print('当前读取的是file1.txt文件')
    else:
        print('当前读取的是file2.txt文件')
    # 处理每一行代码

在上面的代码中，我们可以看到 fileinput.filename() 方法的使用。它用于判断当前处理的是哪个文件。如果是 file1.txt 文件，那就输出相应的信息，否则就输出另一个信息。

注意事项

• 在使用 fileinput 模块处理多个文件时，指定的文件名可以是文件路径，也可以是通配符，比如 *.txt。
• 如果你想指定文件的打开模式，可以使用 fileinput.input(files=('file.txt',), mode='r')。
• 在处理多个文件时，fileinput 会自动将文件拼接起来作为一个输入流，每次处理完一个文件，就会自动关闭它，然后再打开下一个文件。如果你需要保持文件的打开状态，可以设置 inplace=True。