{
"cells": [
{
"cell_type": "markdown",
"metadata": {},
"source": [
"\n",
""
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Lifelines examples\n",
"##### Gehan Survival Dataset\n",
"\n",
"The data show the length of remission in weeks for two groups of leukemia patients, treated and control, and were analyzed by Cox in his original proportional hazards paper. The data are available in a file containing three columns:\n",
"\n",
"- Treatment: coded Treated (drug) or Control (placebo),\n",
"- Time: weeks of remission,\n",
"- Failure: coded 1 if a failure (relapse), 0 if censored"
]
},
{
"cell_type": "code",
"execution_count": 1,
"metadata": {
"collapsed": false
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Populating the interactive namespace from numpy and matplotlib\n"
]
},
{
"name": "stderr",
"output_type": "stream",
"text": [
"/Users/camerondavidson-pilon/.virtualenvs/data/lib/python2.7/site-packages/IPython/core/magics/pylab.py:161: UserWarning: pylab import has clobbered these variables: ['datetime']\n",
"`%matplotlib` prevents importing * from pylab and numpy\n",
" \"\\n`%matplotlib` prevents importing * from pylab and numpy\"\n"
]
}
],
"source": [
"from lifelines.estimation import KaplanMeierFitter, NelsonAalenFitter\n",
"import pandas as pd\n",
"import numpy as np\n",
"%pylab inline\n",
"figsize(12.5,6)"
]
},
{
"cell_type": "code",
"execution_count": 34,
"metadata": {
"collapsed": false
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"/Users/camerondavidson-pilon/.virtualenvs/data/lib/python2.7/site-packages/ipykernel/__main__.py:2: ParserWarning: Falling back to the 'python' engine because the 'c' engine does not support regex separators (separators > 1 char and different from '\\s+' are interpreted as regex); you can avoid this warning by specifying engine='python'.\n",
" from ipykernel import kernelapp as app\n"
]
}
],
"source": [
"data = pd.read_csv('../lifelines/datasets/gehan.dat', sep=\"\\s{1,3}\", \n",
" header=None)\n",
"data.columns = ['treatment', 'time', 'failure']"
]
},
{
"cell_type": "code",
"execution_count": 35,
"metadata": {
"collapsed": false
},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" treatment | \n",
" time | \n",
" failure | \n",
"
\n",
" \n",
" \n",
" \n",
" | 37 | \n",
" 1 | \n",
" 12 | \n",
" 1 | \n",
"
\n",
" \n",
" | 38 | \n",
" 1 | \n",
" 15 | \n",
" 1 | \n",
"
\n",
" \n",
" | 39 | \n",
" 1 | \n",
" 17 | \n",
" 1 | \n",
"
\n",
" \n",
" | 40 | \n",
" 1 | \n",
" 22 | \n",
" 1 | \n",
"
\n",
" \n",
" | 41 | \n",
" 1 | \n",
" 23 | \n",
" 1 | \n",
"
\n",
" \n",
"
\n",
"