import pandas as pd
import rpy2.robjects as ro

df_addicts = pd.read_csv("demo_surv/addicts.csv")

df_addicts.head()

%load_ext rpy2.ipython

%Rpush df_addicts

%%R
library(survival)

Y <- with(df_addicts, Surv(survtime, status == 1))
model_coxph <- coxph(Y ~ strata(clinic) + prison + mdosedata, data=df_addicts)
print(summary(model_coxph))