tracking_tools/dark_light_preference.py
2021-01-21 08:36:35 +01:00

29 lines
778 B
Python

import numpy as np
import os
import glob
import matplotlib.pyplot as pyplot
from IPython import embed
if __name__ == "__main__":
result_files = sorted(glob.glob("./results_day*.npy"))
results = {}
for r in result_files:
day = r.split(os.sep)[-1].split('_')[-1].split('.')[0]
if "day" in results.keys():
results["day"].append(day)
else:
results["day"] = [day]
data = np.load(r, allow_pickle=True)
d = data.item()
for k in d.keys():
if k in results.keys():
results[k].append(d[k])
else:
results[k] =[d[k]]
#res= dict((k, results[k]) for k in ['lepto03DLC' , 'lepto48DLC'] if k in results)
#print(res)
print(results)