Add obs index to label file (#928)

* remove debug logging

* load and save obs/row index in label file

* lint

* update tests
This commit is contained in:
Bruce Martin
2019-09-19 09:16:01 -07:00
committed by GitHub
parent 197ea9214a
commit 109c9e70ec
5 changed files with 23 additions and 26 deletions
+2 -2
View File
@@ -8,7 +8,7 @@ import pandas as pd
def read_labels(fname):
if exists(fname) and getsize(fname) > 0:
return pd.read_csv(fname, dtype='category')
return pd.read_csv(fname, dtype='category', index_col=0)
else:
return pd.DataFrame()
@@ -16,7 +16,7 @@ def read_labels(fname):
def write_labels(fname, df):
rotate_fname(fname)
if not df.empty:
df.to_csv(fname, index=False)
df.to_csv(fname)
else:
open(fname, 'a').close()