remove empty column

tompollard · Jan 29, 2018 · c411e7b · c411e7b
1 parent 8a84244
commit c411e7b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/tableone.py b/tableone.py
@@ -359,6 +359,9 @@ def _create_tableone(self,data):
             dupe_columns.append('ptest')         
         table[dupe_columns] = table[dupe_columns].mask(dupe_mask).fillna('')
 
+        # remove empty column added above
+        table.drop([''], axis=1, inplace=True)
+
         # remove isnull column if not needed
         if not self.isnull:
             table.drop('isnull',axis=1,inplace=True)

diff --git a/test_tableone.py b/test_tableone.py
@@ -197,12 +197,11 @@ def test_sequence_of_cont_table(self):
             categorical = categorical, groupby = groupby, isnull = False)
 
         # n and weight rows are already ordered, so sorting should not alter the order
-        assert (t.tableone.loc['n'].values[0][1:5].astype(float) == \
-            sorted(t.tableone.loc['n'].values[0][1:5].astype(float))).any()
-        assert (t.tableone.loc['age'].values[0][1:5] == \
+        assert (t.tableone.loc['n'].values[0].astype(float) == \
+            sorted(t.tableone.loc['n'].values[0].astype(float))).any()
+        assert (t.tableone.loc['age'].values[0] == \
             ['0.50 (0.71)', '3.50 (1.29)', '8.50 (1.87)', '15.50 (2.45)']).any()
 
-
     @with_setup(setup, teardown)
     def test_categorical_cell_count(self):
         """