def maps_by_group(gens = ['wt', 'ko'], exps = ['nai', 'exp', 'w1', 'w2', 'w3']): for i, (gen, exp) in enumerate(itertools.product(gens, exps)): sesss = glob.glob(os.path.join(basedir, 'Sessions', gen+'_'+exp+'*')) sesss = [os.path.basename(s) for s in sesss] nsess = len(sesss) if nsess > 0: fig, ax = RF.look_at_map(sesss) fig.savefig(os.path.join(basedir, 'maps', '%s_%s' % (gen, exp))) plt.close('all') return