Comparando tratamentos para dor em pacientes

Há um experimento para um novo tratamento de dor, e você quer comparar os pacientes que receberam o novo tratamento com os que receberam o antigo. Você vai ajustar um estimador de Kaplan-Meier para cada conjunto de dados de pacientes e visualizar suas curvas de sobrevivência lado a lado.

O DataFrame se chama recur e contém as colunas:

• age: a idade do paciente;
• treat: o tratamento que o paciente recebeu (0 se novo, 1 se antigo);
• duration: a duração após o tratamento, em horas;
• censor: indica se o evento está censurado;

O pacote pandas está carregado como pd, a classe KaplanMeierFitter foi importada de lifelines, e o módulo pyplot foi importado de matplotlib como plt.

Além disso, uma instância de KaplanMeierFitter chamada kmf e uma figura de subplot chamada ax já foram criadas para você.