python - 创建文件夹并设置为工作目录以保存标准输出文件和图像

Question

在我的代码中，我为 stdout 创建了一个文本文件，还保存了几个 .png 图像和 .mat 矩阵 - 当代码完成运行时，目录中有很多文件

我希望代码能够在运行我的代码的文件夹中创建一个新目录，并将 .txt 文件以及输出 .png 和 .mat 保存到这个新创建的文件夹中。

我已经想出要创建我应该做的新目录：

import os
os.mkdir('folder')

并创建输出文件并将标准输出设置在那里

import sys
filename = open('filename.txt','w')
sys.stdout = filename

我尝试使用 open('folder/filename.txt','w') 但出现错误：IOError: [Errno 2] No such file or directory

谢谢！

score 1 · Accepted Answer

如果我理解正确，您想在刚刚创建的文件夹（“文件夹”）中创建文件“filename.txt”吗？

鉴于这种情况，请使用 os.path.join()

import sys
filename = open(os.path.join('folder','filename.txt'),'w')
sys.stdout = filename

现在 sys.stdout 指向新文件夹中的文件

1 回答 1