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()
