如何保存lifelines模块绘制的图片

Question

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

kmf这个工具没有保存图像的功能。当我运行'kmf.plot()'的时候，图像只会在命令行窗口显示出来。我希望kmf能像pyplot.savefig那样，这样我就可以把图像保存到电脑上了。

Answer 1

lifelines模块使用matplotlib来绘制图表。plot方法会返回一个AxesSubplot实例，这个实例代表了绘制的图表。你可以用这个实例来访问包含这个子图的整个图形，然后就可以使用savefig方法来保存图形：

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