Commit 937e4442 authored by Joanne Hugé's avatar Joanne Hugé

Remove useless color array

parent 03d17864
......@@ -134,9 +134,6 @@ class MeasureSetHandler:
def generate_graphs(self, metadata_masks):
# List of colors to use to superimpose the different plots
colors = ['red', 'blue', 'green', 'purple', 'yellow']
# For each type of measure set
for mtype in self.measure_sets:
......@@ -169,9 +166,9 @@ class MeasureSetHandler:
metadata_infos.append(", ".join(metadata_info))
if props_type == 'histogram':
subplots.append(measure.generate_histogram(i, colors[j], metadata_info))
subplots.append(measure.generate_histogram(i, metadata_info))
else:
subplots.append(measure.generate_chrono_graph(i, colors[j], metadata_info))
subplots.append(measure.generate_chrono_graph(i, metadata_info))
fig, ax = plt.gcf(), plt.gca()
......@@ -501,7 +498,7 @@ class MeasureSet:
self.add_chronological(props_names, props)
def generate_histogram(self, i, color, name):
def generate_histogram(self, i, name):
bins = [i for i in range(self.min[i], self.max[i] + 1)]
vals = self.props[i][self.min[i] + self.middle : self.max[i] + self.middle + 1]
......@@ -513,10 +510,10 @@ class MeasureSet:
return plt.bar(bins, vals, alpha=0.4)
def generate_chrono_graph(self, i, color, name):
def generate_chrono_graph(self, i, name):
prop = self.props[i]
x = [i for i in range(len(prop))]
return plt.plot(x, prop, color=color)
return plt.plot(x, prop)
def generate_table(self, headers=True, values=True, metadata_mask=[], props_lens=[]):
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment