我正在使用lifelines library学习生存分析。我正在使用leukemia dataset。我正在尝试绘制寿命图。但是,我得到的图是颠倒的(即垂直翻转)。
我正在使用此notebook作为参考。这是我的代码:
import lifelines
import matplotlib.pyplot as plt
from lifelines.datasets import load_leukemia
leukemia = load_leukemia()
durations = leukemia.t.values
events = leukemia.status.values
ax = lifelines.plotting.plot_lifetimes(durations=durations, event_observed=events)
ax.set_xlim(0, 40)
ax.set_xlabel("Time in Months")
plt.plot()
这是我得到的情节
这是我要生成的情节:
答案 0 :(得分:0)
尽管我不确定哪个图是正确的,但是lifelines
API已于2019年3月21日更新(v。0.20.2)。按照changelog:
plot_lifetimes在绘制时不再颠倒顺序。
github笔记本的日期为2019年3月11日,因此我认为它使用的是旧API,该旧API可能导致图中所示的绘图。但是,lifelines
的所有将来版本都会导致您得到该绘图。
因此,基本上,如果您卸载当前的lifelines
程序包和pip install lifelines==0.24.13
(最新的lifelines
版本为0.24.13),您将得到自己想要的图。
Cameron Davidson(lifelines
的作者)也许可以对此有所了解-可能是在更改API之后需要更新白血病数据集!