如何用不同的名称保存文件而不覆盖

Question

我需要在不同的数据集上重复运行一个python脚本。python脚本test.py使用命令处理数据集、打印和保存结果。

plt.savefig('result.png')

如何确保相同的test.py脚本在另一个数据集上运行。。新的result.png不会覆盖我以前的结果吗？基本上，在执行plt.savefig('result.png')之前，我需要检查result.png是否已经存在，如果已经存在，则将结果重命名为任何其他名称，如

result1.png
result2.png

否则，在下一个后期处理中，文件将被覆盖。

Answer 1

您可以使用标准库中的tempfile.mkstemp

import tempfile
fi, filename = tempfile.mkstemp(prefix='result_', suffix='.png')

注意：fi是一个整数。如果你需要一个文件对象

f = os.fdopen(fi, "w")

而filename包含所创建文件的绝对路径名，例如

'/tmp/result_d_3787js.png'

另一个解决方案是使用uuid，例如

import uuid
filename = 'result_'+str(uuid.uuid4())+'.png'

会产生一些

result_cf29d123-271e-4899-b2f6-d172f157af65.png

更多信息和参数请参见官方文件 在mkstemp和uuid上