experiment.py 文件源码

def load_stage_binned(strain = 'n2', wid  = all, dtype = 'speed', nbins_per_stage = 10, function = np.nanmean, nout = 1):
  """Load data an bin at different stages"""
  if isinstance(wid, int):
    wids = [wid];
  else:
    wids = wid;

  data = load(strain = strain, wid = wids, dtype = dtype);
  sbins = stage_bins(strain = strain, wid = wids, nbins_per_stage = nbins_per_stage);
  bdata = bin_data(data, sbins, function = function, nout = nout);

  if isinstance(wid, int):
    bdata = bdata[0];

  return bdata;



############################################################################
### Accessing aligned data
############################################################################