for k in range(len(celltype)): ind = 0 ind1 = len(indices) fig = plt.figure(figsize=[8, 8]) for i in range(len(indices)): slopes.append([]) pars = indices[i] par1['CD'] = cd[pars[0]] par1['g1_thresh_std'] = g1_std[pars[1]] par1['g2_std'] = g2_std[pars[2]] par1['l_std'] = l_std[pars[3]] if k == 0: # don't do it twice for j in range(len(altmodels)): slopes[i].append([]) par1['modeltype'] = altmodels[j] obs_temp, tg_temp = g.single_par_meas4(par1) # discretized gen slopes[i][j].append( obs_temp[0, :]) # mothers first, then daughters frac = np.array(1 - 2**(-par1['CD'])) for j in range(len(models)): if j == 0: ind += 1 index = ind if j == 1: ind1 += 1 index = ind1 colorval = scalarmap.to_rgba(values[index]) plt.plot(f, obs[pars[0], pars[1], pars[2], pars[3], :, j, 0, k], label=modeltype[j] + ' $\sigma_i$=' + str(np.round(par1['g1_thresh_std'], 2)) +
#a = np.zeros((X, Y, Z, 6, 2, 3)) if rank == 0: print 'Setup matrix' dx = X / size start = dx * rank stop = start + dx if rank == size - 1: stop = X for i in xrange(start, stop): par1['g2_std'] = g2_std[i] for j in range(Y): par1['g1_thresh_std'] = g1_std[j] for k in range(Z): par1['d_std'] = d_std[k] # obs1, obs2, tg1, tg2 = g.single_par_meas5(par1) # Obs1 are leaf cells, obs2 are entire tree cells obs3, tg3 = g.single_par_meas4(par1) a[i, j, k, :, :] = obs3[:6, :] basepath = '/home/felix/simulation_data/discr_time_tester1_data/' # np.save(basepath + 'tgrow_m_model'+str(par1['modeltype'])+'_discr_time_leaf_cd_' # + str(i) + '_s1_' + str(j) + '_s2_' + str(k), tg1[0]) # np.save(basepath + 'tgrow_m_model'+str(par1['modeltype'])+'_discr_time_tree_cd_' # + str(i) + '_s1_' + str(j) + '_s2_' + str(k), tg2[0]) np.save( basepath + 'tgrow_m_model' + str(par1['modeltype']) + '_discr_genr_tree_cd_' + str(i) + '_s1_' + str(j) + '_s2_' + str(k), tg3[0]) # np.save(basepath + 'tgrow_d_model'+str(par1['modeltype'])+'_discr_time_leaf_cd_' # + str(i) + '_s1_' + str(j) + '_s2_' + str(k), tg1[1]) # np.save(basepath + 'tgrow_d_model'+str(par1['modeltype'])+'_discr_time_tree_cd_' # + str(i) + '_s1_' + str(j) + '_s2_' + str(k), tg2[1]) np.save(
stop = X for i in xrange(start, stop): par1['CD'] = cd[i] for j in range(Y): par1['g1_thresh_std'] = g1_std[j] for k in range(Z): par1['g2_std'] = g2_std[k] for m in range(W): par1['l_std'] = l_std[m] for n in range(V): par1['frac'] = f[n] for num in range(len(models)): par1['modeltype'] = models[num] # obs1, obs2, tg1, tg2 = g.single_par_meas5(par1) # Obs1 are leaf cells, obs2 are entire tree cells obs3, tg3 = g.single_par_meas4( par1) # discretized gen a[i, j, k, m, n, num, :, :] = obs3[:6, :] # discr gen # a[i, j, k, m, n, num, :, :] = obs2[:6, :] # discr time tree # a[i, j, k, m, n, num, 2, :, :] = obs1[:6, :] # discr time leaf basepath = '/home/felix/simulation_data/model11_12/' # np.save(basepath + 'tgrow_m_model'+str(par1['modeltype'])+'_discr_time_leaf_cd_' # + str(i) + '_s1_' + str(j) + '_s2_' + str(k) + '_sl_' + str(m) + '_f_' + str(n), # tg1[0]) # np.save(basepath + 'tgrow_m_model'+str(par1['modeltype'])+'_discr_time_tree_cd_' # + str(i) + '_s1_' + str(j) + '_s2_' + str(k) + '_sl_' + str(m) + '_f_' + str(n), # tg2[0]) np.save( basepath + 'tgrow_m_model' + str(par1['modeltype']) + '_discr_genr_tree_cd_' + str(i) + '_s1_' +