Python中FileInput()函数的常见问题和解决方法

在Python中，fileinput模块提供了一个FileInput()函数，可以方便地处理多个文件的输入。它允许你在for循环中遍历多个文件的行，而无需在处理每个文件之间打开和关闭文件。

然而，在使用FileInput()函数的过程中，可能会遇到一些常见问题。下面是几个可能的问题以及解决方法和使用例子：

1. 问题：如何遍历文件中的每一行？

解决方法：可以使用for循环迭代FileInput()对象，它会自动返回每个文件的行。

import fileinput

for line in fileinput.input():
    print(line)

2. 问题：如何获取当前处理的文件的文件名？

解决方法：可以使用fileinput.filename()函数获取当前处理的文件的文件名。

import fileinput

for line in fileinput.input():
    print(fileinput.filename())

3. 问题：如何判断当前处理的行是来自于哪个文件？

解决方法：可以使用fileinput.filelineno()函数获取当前处理的行号，使用fileinput.filelineno()函数获取当前处理的文件的文件名。

import fileinput

for line in fileinput.input():
    print(f"Line {fileinput.filelineno()} in {fileinput.filename()}: {line}")

4. 问题：如何修改文件中的内容？

解决方法：可以使用fileinput.input()函数返回的迭代器的fileinput.filename()和fileinput.filelineno()函数获取当前处理的文件和行号，然后在需要修改的地方进行操作。

import fileinput

for line in fileinput.input(inplace=True):
    if fileinput.filename() == 'file.txt' and fileinput.filelineno() == 3:
        print("Updated line")
    else:
        print(line, end='')

在上面的例子中，如果当前处理的文件是'file.txt'并且当前处理的行号是3，那么会将"Updated line"写入到该行。

以上是FileInput()函数的常见问题和解决方法带有使用例子。通过使用这些技巧，你可以更方便地处理多个文件的输入，并进行必要的操作和修改。