如何保存生命线模块绘制的图片

Question

In [3]: type(kmf) 
Out[3]: lifelines.estimation.KaplanMeierFitter
In [2]: kmf.plot()
Out[2]: <matplotlib.axes.AxesSubplot at 0x533b970>

kmf没有保存图形的功能？当我运行'kmf.plot（）'时，图片会显示在shell中，我需要kmf可以像pyplot.savefig，然后我可以将图形保存到磁盘。

Answer 1

该lifelines模块matplotlib用于绘图。该plot方法返回AxesSubplot绘图的一个实例。这可用于访问包含子图的图形，然后savefig可以使用该方法：

ax = kmf.plot()
ax.get_figure().savefig("myfigure.png")