用Python实现生存曲线图

生存曲线图是生存分析中常用的一种图形展示方式，Python中可以使用 lifelines 库来实现生存曲线图的绘制。具体步骤如下：

1.安装 lifelines 库

!pip install lifelines

2.导入所需库和数据

import pandas as pd
from lifelines import KaplanMeierFitter
import matplotlib.pyplot as plt

# 读取数据
data = pd.read_csv('survival_data.csv')

3.对数据进行预处理

# 将时间和事件列提取出来
time = data['time']
event = data['event']

# 将事件列中的0和1替换为False和True
event = event.replace({0: False, 1: True})

4.使用 KaplanMeierFitter 计算生存曲线数据

# 创建 KaplanMeierFitter 对象
kmf = KaplanMeierFitter()

# 使用 fit() 方法计算生存曲线数据
kmf.fit(time, event)

5.绘制生存曲线图

# 使用 plot() 方法绘制生存曲线图
kmf.plot()

# 添加图例、坐标轴标签等
plt.title('Survival Curve')
plt.xlabel('Time')
plt.ylabel('Survival Probability')
plt.show()