阅读并做一些事情

Question

我如何遍历目录？我这样做是为了在诗人的张量流中吐出预测。它通常只接受一个文件并吐出预测。 我打算给它整个目录，并在一些文本文件中保存预测。我通过docker接口这样做，我想写一个脚本，它将逐个遍历目录中的所有文件并存储预测在一些文本文件中。

任何帮助都将不胜感激。

Answer 1

我在编写代码时遇到了类似的问题。 以下代码将递归文件夹中的所有文件（包括子文件夹）。

import os

with open('prediction.txt','w') as tf_file:
    for dirPath, dirNames, fileNames in os.walk(folder_name):
        for fileName in fileNames:
            results = perform_tensorflow_stuff(fileName)
            tf_file.writelines('\n'.join(results))

希望这有帮助！