Added Sciatica trial data set and Leukemia trial datasets

zepid/__init__.py

@@ -36,7 +36,7 @@
 """
 from .base import (RiskRatio, RiskDifference, NNT, OddsRatio, IncidenceRateRatio, IncidenceRateDifference, Sensitivity,
                    Specificity, Diagnostics, interaction_contrast, interaction_contrast_ratio, spline, table1_generator)
-from .datasets import load_sample_data, load_ewing_sarcoma_data, load_gvhd_data, load_sciatica_data
+from .datasets import load_sample_data, load_ewing_sarcoma_data, load_gvhd_data, load_sciatica_data, load_leukemia_data
 
 import zepid.calc
 import zepid.graphics

zepid/datasets/__init__.py

@@ -35,7 +35,7 @@ def load_sample_data(timevary):
     if timevary is True:
         return df
     else:
-        dfi = df.loc[df.groupby('id').cumcount() == 0][['id', 'male', 'age0', 'cd40', 'cd4', 'dvl0', 'art']].copy()
+        dfi = df.loc[df.groupby('id').cumcount() == 0][['id', 'male', 'age0', 'cd40', 'dvl0', 'art']].copy()
         dfo = df.loc[df.id != df.id.shift(-1)][['id', 'dead', 'drop', 'out']].copy()
         dfo.loc[(dfo['drop'] == 1) & (dfo['out'] <= 45), 'dead'] = np.nan
         dfo['dead'] = np.where((dfo['dead'] == 1) & (dfo['out'] > 45), 0, dfo['dead'])
@@ -190,3 +190,19 @@ def load_sciatica_data():
                                          'likert_t', 'vas1_b', 'vas2_b', 'roland_b', 'likert_b', 'male', 'weight',
                                          'height', 'surgery'])
     return df
+
+
+def load_leukemia_data():
+    """Loads data from Freireich EJ et al., "The Effect of 6-Mercaptopurine on the Duration of Steriod-induced
+    Remissions in Acute Leukemia: A Model for Evaluation of Other Potentially Useful Therapy" Blood 1963
+
+    Variables:
+    -t: time
+    -status: event indicator (0: censored, 1: relapsed)
+    -sex: male, female
+    -logwbc: log-transformed white blood cell count
+    -treat: treatment indicator
+    """
+    df = pd.read_csv(resource_filename('zepid', 'datasets/leukemia.dat'), delim_whitespace=True, index_col=False)
+    return df
+