def change_ply_colors(dpv_file, colors_values): colors = utils.arr_to_colors(colors_values) start_reading = False with open(dpv_file, 'r+') as f: for line in f.readlines(): if (line.strip() == 'end_header'): start_reading = True continue if (start_reading): vals = line.strip().split(' ')
def change_ply_colors(dpv_file, colors_values): colors = utils.arr_to_colors(colors_values) start_reading = False with open(dpv_file, "r+") as f: for line in f.readlines(): if line.strip() == "end_header": start_reading = True continue if start_reading: vals = line.strip().split(" ")
def scatter_plot_perm_ttest_results(points, values, fs_pts, max_vals, fol): fig = plt.figure() ax = Axes3D(fig) ax.scatter(fs_pts[0, :, 0], fs_pts[0, :, 1], fs_pts[0, :, 2], c='white', edgecolors='none', alpha=0.01) ax.scatter(fs_pts[1, :, 0], fs_pts[1, :, 1], fs_pts[1, :, 2], c='white', edgecolors='none', alpha=0.01) m = utils.get_scalar_map(1, max_vals, color_map='YlOrRd') for ind, t in enumerate(points.keys()): print(t) colors = utils.arr_to_colors(values[t], 1, max_vals, colors_map='YlOrRd') sca = ax.scatter(points[t][:, 0], points[t][:, 1], points[t][:, 2], c=colors) if ind == 0: m.set_array(colors) plt.colorbar(m, ticks=range(1, max_vals + 1), shrink=.5, aspect=10) title_obj = plt.title('t={}'.format(t)) else: plt.setp(title_obj, text='t={}'.format(t)) plt.savefig(op.join(fol, '{}.jpg'.format(t))) sca.remove() plt.close()