if __name__ == "__main__": from argparse import ArgumentParser from collections import namedtuple from ptsa.data.readers import JsonIndexReader os.environ['RHINO_ROOT'] = "~/mnt/rhino" Dataset = namedtuple("Dataset", "subject,experiment,session") parser = ArgumentParser() parser.add_argument("outfile", type=str, help="path to output HDF5 file") args = parser.parse_args() reader = JsonIndexReader('/Users/depalati/mnt/rhino/protocols/r1.json') df = reader.as_dataframe().reset_index() subjects = df[df.experiment.isin(['FR5', 'catFR5'])].subject.unique() datasets = [ # Dataset('R1395M', "catFR5", 4), ] for subject in subjects: for session in range(4): datasets.append(Dataset(subject, 'FR5', session)) datasets.append(Dataset(subject, 'catFR5', session)) with h5py.File(args.outfile, 'w') as hfile: count = 0 for dataset in datasets:
def test_as_dataframe(self): index = osp.join(get_rhino_root(), "protocols", "r1.json") reader = JsonIndexReader(index) df = reader.as_dataframe() assert isinstance(df, pd.DataFrame)