Metadata-Version: 2.1
Name: survival_data_handler
Version: 2024.4.5
Author-email: Vincent Laurent <vlaurent@eurobios.com>
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: matplotlib>=3
Requires-Dist: seaborn
Requires-Dist: numba
Requires-Dist: scipy
Requires-Dist: scikit-learn
Requires-Dist: pandas>2
Requires-Dist: lifelines
Requires-Dist: numpy

# Survival data handler
![code coverage](https://raw.githubusercontent.com/eurobios-mews-labs/survival-data-handler/coverage-badge/coverage.svg?raw=true)
[![PyPI version](https://badge.fury.io/py/palma.svg)](https://badge.fury.io/py/palma)

```python
import pandas as pd

from lifelines import CoxPHFitter
from lifelines.datasets import load_rossi

rossi = load_rossi()
cph = CoxPHFitter()

cph.fit(rossi, duration_col='week', event_col='arrest')
curves = cph.predict_survival_function(rossi).T
curves.columns = pd.to_timedelta(curves.columns.to_numpy() * 7, unit="D")
print(curves.head())
```

|    |   7 days 00:00:00 |   14 days 00:00:00 |   21 days 00:00:00 |   28 days 00:00:00 |   35 days 00:00:00 |   42 days 00:00:00 |   49 days 00:00:00 |   56 days 00:00:00 |   63 days 00:00:00 |   70 days 00:00:00 |
|---:|------------------:|-------------------:|-------------------:|-------------------:|-------------------:|-------------------:|-------------------:|-------------------:|-------------------:|-------------------:|
|  0 |          0.997616 |           0.99523  |           0.992848 |           0.990468 |           0.988085 |           0.985699 |           0.983305 |           0.971402 |           0.966614 |           0.964223 |
|  1 |          0.993695 |           0.987411 |           0.981162 |           0.974941 |           0.968739 |           0.962552 |           0.95637  |           0.926001 |           0.913958 |           0.907978 |
|  2 |          0.994083 |           0.988183 |           0.982314 |           0.976468 |           0.970639 |           0.96482  |           0.959004 |           0.930402 |           0.919043 |           0.913399 |
|  3 |          0.999045 |           0.998089 |           0.997133 |           0.996176 |           0.995216 |           0.994254 |           0.993287 |           0.98846  |           0.986508 |           0.985531 |
|  4 |          0.997626 |           0.99525  |           0.992878 |           0.990507 |           0.988135 |           0.985758 |           0.983374 |           0.97152  |           0.966752 |           0.96437  |

