Python：将来自不同目录的文件输出附加到一个整体列表中

Question

我有 n 个目录（标记为 0 到 n），每个目录都有一个文件（所有文件都具有相同的名称），我想从中获取每个文件中的某些行。然后我想将这些抓取的行按顺序（从 0 到 n）附加到一个列表中。

这是我的设置：

for i in range(0, nfolders):
    folder = "%02d" % i
    os.system("cd " + folder)

    myFile = open("myOutputFile", "r")
    lines = myFile.readlines()
    firstLine = float(lines[0])

    #I then write a loop to store the next 5 lines in a list using append and call this list nextLines

我的问题是，有没有一种简单的方法可以将所有目录中的 firstLine 附加到一个列表中（我的函数返回），以及将所有目录中的 nextLines 附加到一个列表中（同样，我的函数返回）？

我知道有扩展功能，我可以在这里循环吗（因为假设我有 nfolders = 300，很难手动添加东西）？

谢谢！

Answer 1

你有几个问题要处理。 os.system 更改子shell 调用的工作目录（然后立即退出），但不更改此运行脚本的目录。为此使用os.chdir。或者，更好的是，只需将路径添加到文件名并使用它。

您无需阅读整个文件即可获得其第一行，.readline 或 next() 函数会为您完成这项工作。最后，只需附加到一个列表。

my_list = []

for i in range(0, nfolders):
    filename = "%02d/MyOutputFile" % i
    with open(filename) as myFile:
        firstLine = float(next(myFile))
        my_list.append(firstLine)

更新

假设您想要每个文件中的4 + i 行。你可以把它收紧

my_list = []

for i in range(0, nfolders):
    filename = "%02d/MyOutputFile" % i
    with open(filename) as myFile:
        my_list += (next(myFile) for _ in range(4+i))

请注意，我们只使用range 来计算迭代次数，并不关心它的值，因此我们使用变量_ 作为不需要该值的快速可视队列。