创建文件夹并设置为工作目录以保存stdout文件和图像

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]没有这样的文件或目录

谢谢！

Answer 1

如果我理解你的话，你想在你刚刚创建的文件夹（'folder'）中创建'filename.txt'文件吗？

鉴于这种情况，请使用os.path.join（）

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

现在sys.stdout指向新文件夹

内的文件