Python 将Plot和Print语句保存在同一文件中

在过去的几天里，我一直在编写一个程序，该程序可以绘制多个不同的绘图，并为每个绘图编写一个打印语句来描述特定的条件。循环将保存每个绘图，然后再进行下一次迭代。所有打印均成功保存，但未与其关联的打印语句一起保存

我的代码：

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()

如果ab：
打印（'A大于B'）
ax.set_title（'事件编号{}，事件类型：{}，事件结果：{}'。格式（当前事件编号，当前事件类型，当前事件结果））
sns.散点图（x，y，色调=颜色，图例=假）
label_point（x，y，start_loc.Name，ax）#用于标记散点图的函数
sns.线形图（x1，y1，颜色2，图例=假，ci=无）
plt.绘图（x2，y2，颜色为红色，标记为o）
savefig（'Event{}.png'.format（i），format='png'）
plt.show（）
plt.close（）

---

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

打印到输出，打印到PNG。如果你想打印到PNG，你可以在标题或类似的地方，因为你不能在PNG中输出文本，它只用于图形。相反，您可以创建一个日志文件。在日志文件中进行所有打印，并写入PNG的文件名。然后，所有内容都记录在日志中，您可以在以后需要时连接文件。要在日志中打印，请使用write。

请阅读并张贴您的答案您的答案更好吗？你也可以回答。