Skip to content
Snippets Groups Projects

correct the way to identify colors for mass pos

Merged Lan Dam requested to merge i157_fix_mass_pos_color into master
1 unresolved thread
@@ -385,18 +385,16 @@ class Plotting:
colors = [None] * len(y)
sizes = [1.5] * len(y)
for i in range(len(y)):
count = 0
prev_v = 0
count = -1
for v, c in value_colors:
if count < (len(value_colors) - 1):
if prev_v < abs(y[i]) <= v:
count += 1
if count <= len(value_colors) - 2:
if abs(y[i]) <= v:
colors[i] = clr[c]
break
else:
# The last value color
colors[i] = clr[c]
break
prev_v = v
count += 1
ax.scatter(x, y, marker='s', c=colors, s=sizes,
zorder=constants.Z_ORDER['DOT'])
ax.x_list = x_list
Loading