如何在 lifelines Kaplan-Meier 图中使用自定义图例标签？
使用 lifelines 库，你可以轻松绘制 Kaplan-Meier 图，例如在我们之前的文章最小 Python Kaplan-Meier 图示例中所见：
kaplan_meier_example.py
from lifelines.datasets import load_leukemia
from lifelines import KaplanMeierFitter
df = load_leukemia()
kmf = KaplanMeierFitter()
kmf.fit(df['t'], df['Rx']) # t = 时间点, Rx: 0=删失, 1=事件
kmf.plot()
如果你想要自定义标签而不是 KM_estimates 出现在图例中怎么办？
使用 kmf.fit(..., label='Your label')。由于我们在此示例中使用 leukemias 数据集，我们使用标签 'Leukemia'
完整示例：
custom_label_example.py
from lifelines.datasets import load_leukemia
from lifelines import KaplanMeierFitter
# 加载数据集
df_leukemia = load_leukemia()
# 拟合并绘制白血病数据集
kmf_leukemia = KaplanMeierFitter()
kmf_leukemia.fit(df_leukemia['t'], df_leukemia['Rx'], label="Leukemia")
ax = kmf_leukemia.plot()
# 将 Y 轴设置为固定比例
ax.set_ylim([0.0, 1.0])
Check out similar posts by category:
Python, Statistics
If this post helped you, please consider buying me a coffee or donating via PayPal to support research & publishing of new posts on TechOverflow