python - enumerate(fileinput.input(file)) 和 enumerate(file) 的区别

Question

我正在为我的代码寻求一些帮助，如下所示:

for file in file_name :
    if os.path.isfile(file):
        for line_number, line in enumerate(fileinput.input(file, inplace=1)):
            print file
            os.system("pause")
            if line_number ==1:
                line = line.replace('Object','#Object')
                sys.stdout.write(line)

我想修改一些以前提取的文件，以便用 matplotlib 绘制它们。为此，我删除了一些行，评论了一些其他行。

我的问题如下:

• 使用 for line_number, line in enumerate(fileinput.input(file, inplace=1)): 只给我 5 个先前提取的文件中的 4 个(当查看 file_name 列表时包含 5引用!)

• 使用 for line_number, line in enumerate(file): 给我之前提取的 5 个文件，但我不知道如何在不创建另一个文件的情况下使用同一个文件进行修改...

您对这个问题有什么想法吗？这是正常问题吗？

Answer 1

有许多事情可能对您有所帮助。

首先 file_name 似乎是一个文件名列表。将它命名为 file_names 可能更好，然后您可以为每个文件使用 file_name。您已验证这确实包含 5 个条目。

enumerate() 函数用于在枚举项目列表时提供帮助，以便为每个循环提供索引和项目。这使您不必使用单独的计数器变量，例如

for index, item in enumerate(["item1", "item2", "item3"]):
    print index, item

将打印:

0  item1
1  item2
2  item3

这并不是真正需要的，因为您已选择使用 fileinput 库。这旨在获取文件列表并在一个循环中遍历所有文件中的所有行。因此，您需要稍微调整一下您的方法，假设您的文件列表被称为 file_names 然后您编写如下内容:

# Keep only files in the file list
file_names = [file_name for file_name in file_names if os.path.isfile(file_name)]

# Iterate all lines in all files
for line in fileinput.input(file_names, inplace=1):
    if fileinput.filelineno() == 1:
        line = line.replace('Object','#Object')
        sys.stdout.write(line)  

这里的要点是最好在将列表传递给 fileinput 之前预过滤任何非文件名。我将由您来修复输出。

fileinput 提供了许多函数来帮助您确定当前正在处理的文件或行号。