列线图Python代码

以下是使用Python绘制Nomogram列线图的示例代码：

import matplotlib.pyplot as plt
from sklearn.preprocessing import LabelEncoder
from sksurv.preprocessing import OneHotEncoder
from sksurv.datasets import load_veterans_lung_cancer
from sksurv.linear_model import CoxPHSurvivalAnalysis
from sksurv.util import Surv
import pandas as pd

# 加载数据集
data_x, y = load_veterans_lung_cancer()
x = pd.DataFrame(data_x)

# 对分类变量进行编码
categorical_columns = ['Celltype', 'Karno', 'Diagtime', 'Age', 'Prior']
x_enc = x.copy()
for col in categorical_columns:
    le = LabelEncoder()
    x_enc[col] = le.fit_transform(x[col])

# 使用OneHotEncoder对分类变量进行独热编码
ohe = OneHotEncoder()
x_enc_ohe = ohe.fit_transform(x_enc)

# 训练Cox比例风险模型
estimator = CoxPHSurvivalAnalysis()
estimator.fit(x_enc_ohe, Surv.from_arrays(y["Status"], y["Survival_in_days"]))

# 绘制Nomogram列线图
from sksurv.linear_model import plot_nomogram
fig, ax = plt.subplots(figsize=(8, 6))
plot_nomogram(estimator, x_enc_ohe.columns, ax=ax)
plt.show()