('results', '_v_attrs', 'STS')) IDX = get_all_attrs(DB, IDX_ATTRS) IDX.sort() X_IDX = list(set([i[0] for i in IDX])) X_IDX.sort() Y_IDX = list(set([i[1] for i in IDX])) Y_IDX.sort() IMSHOW_EXTENT = (X_IDX[0], X_IDX[-1], Y_IDX[0], Y_IDX[-1]) Z_IDX = [] for i in xrange(6): Z_IDX.append(np.zeros((len(X_IDX), len(Y_IDX)))) for ind_rate in xrange(len(X_IDX)): for ind_strength in xrange(len(Y_IDX)): tmp_mps = IDX[to1d(ind_rate, ind_strength, len(X_IDX))][2] Z_IDX[0][ind_rate][ind_strength] = tmp_mps[0] Z_IDX[1][ind_rate][ind_strength] = tmp_mps[1] Z_IDX[2][ind_rate][ind_strength] = tmp_mps['whole'] tmp_sts = IDX[to1d(ind_rate, ind_strength, len(X_IDX))][3] Z_IDX[3][ind_rate][ind_strength] = tmp_sts[0] Z_IDX[4][ind_rate][ind_strength] = tmp_sts[1] Z_IDX[5][ind_rate][ind_strength] = tmp_sts['whole'] # MPS plotting MPS_FIG, MPS_AXS = plt.subplots(1, 3, figsize=(9, 3)) MPS_MIN_MAX = get_all_min_max([Z_IDX[i] for i in xrange(3)]) MPS_NORM = colors.normalize(MPS_MIN_MAX[0], MPS_MIN_MAX[1]) plot_run(MPS_FIG, "MPS", MPS_AXS, Z_IDX, range(3), MPS_NORM, IMSHOW_EXTENT)
def get_1d_ind_simu(ind, width): """Return 1d index given `ind` like 5,3""" x, y = ind.split(",") x, y = int(x), int(y) return utils.to1d(x, y, width)