将绘图和打印语句保存在同一文件中

Question

过去几天我一直在编写一个程序，该程序制作了几个不同的绘图，并为每个描述特定条件的绘图编写了一个打印语句。在进入下一次迭代之前，循环会保存每个图。这些图都成功保存，但没有与其关联的打印语句。

我的代码：

if a < b:
  print('A is less than B')
if a > b:
  print('A is greater than B')

ax.set_title('Event Number {}, Event Type: {}, Event Outcome: {}'.format(current_event_number, current_event_type, current_event_outcome))
sns.scatterplot(x, y, hue = color, legend= False)
label_point(x, y, start_loc.Name, ax) #Function to label scatterplot
sns.lineplot(x1,y1, color2, legend=False, ci = None)
plt.plot(x2, y2, color='red', marker='o') 
plt.savefig('Event{}.png'.format(i), format='png')
plt.show()
plt.close()

这段代码都是同一个循环的一部分，当它运行时，条件打印语句显示，它们只是不与数字一起保存。我该如何解决这个问题？

Answer 1

打印到输出，绘图到 PNG。如果您想打印到 PNG，您可以输入标题或类似内容，因为您无法在 PNG 中输出文本，它仅用于图形。相反，您可以制作一个日志文件。在日志文件中，您执行所有打印操作，并且还写入 PNG 的文件名。然后一切都记录在日志中，您可以稍后在需要时连接文件。要在日志中打印，请使用 write。