-
Notifications
You must be signed in to change notification settings - Fork 14
/
mpl_graph_line.py
48 lines (40 loc) · 2.2 KB
/
mpl_graph_line.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import matplotlib.pyplot as plt
class mpl_graph_line(object):
def __init__(self, window_title, xlab, ylab, save_img):
# These are the "Tableau 20" colors as RGB.
self.colours_tableau20 = [(31, 119, 180), (174, 199, 232), (255, 127, 14), (255, 187, 120),
(44, 160, 44), (152, 223, 138), (214, 39, 40), (255, 152, 150),
(148, 103, 189), (197, 176, 213), (140, 86, 75), (196, 156, 148),
(227, 119, 194), (247, 182, 210), (127, 127, 127), (199, 199, 199),
(188, 189, 34), (219, 219, 141), (23, 190, 207), (158, 218, 229)]
# Scale the RGB values to the [0, 1] range, which is the format matplotlib accepts.
for i in range(len(self.colours_tableau20)):
r, g, b = self.colours_tableau20[i]
self.colours_tableau20[i] = (r / 255., g / 255., b / 255.)
self.save_img = save_img
self.fig = plt.figure(facecolor='white', dpi=70, figsize=(20, 12))
self.fig.canvas.set_window_title(window_title)
self.ax = self.fig.add_subplot(1, 1, 1)
plt.xlabel(xlab, color='0.4')
plt.ylabel(ylab, color='0.4')
self.prettify()
def prettify(self):
'''Remove Top/Right Lines and Ticks'''
self.ax.spines['right'].set_visible(False)
self.ax.spines['top'].set_visible(False)
self.ax.spines['bottom'].set_color('0.5')
self.ax.spines['left'].set_color('0.5')
self.ax.tick_params(axis='x', which='both', bottom='on', top='off', color='0.5', labelcolor='0.4')
self.ax.tick_params(axis='y', which='both', left='on', right='off', color='0.5', labelcolor='0.4')
def plot(self, data_list,legend_list):
colour_i = 0
for data, lw in data_list:
self.ax.plot(data, linewidth=lw, color=self.colours_tableau20[colour_i])
colour_i += 1
if colour_i >= len(self.colours_tableau20):
colour_i = 0
plt.legend(legend_list, loc='upper left', prop={'size':'10'}, frameon=False)
if self.save_img:
dir_output = 'output'
plt.savefig(dir_output + '/graph_output.png')
plt.show()