-
Notifications
You must be signed in to change notification settings - Fork 299
/
plot_variance_subplots_ter.py
103 lines (80 loc) · 3.02 KB
/
plot_variance_subplots_ter.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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
import glob
from pylab import *
# import colors
from palettable.colorbrewer.qualitative import Set2_7
colors = Set2_7.mpl_colors
params = {
'axes.labelsize': 8,
'font.size': 8,
'legend.fontsize': 10,
'xtick.labelsize': 10,
'ytick.labelsize': 10,
'text.usetex': False,
'figure.figsize': [7, 4]
}
rcParams.update(params)
def load(dir):
f_list = glob.glob(dir + '/*/*/bestfit.dat')
num_lines = sum(1 for line in open(f_list[0]))
i = 0
data = np.zeros((len(f_list), num_lines))
for f in f_list:
data[i, :] = np.loadtxt(f)[:, 1]
i += 1
return data
def perc(data):
median = np.zeros(data.shape[1])
perc_25 = np.zeros(data.shape[1])
perc_75 = np.zeros(data.shape[1])
for i in range(0, len(median)):
median[i] = np.median(data[:, i])
perc_25[i] = np.percentile(data[:, i], 25)
perc_75[i] = np.percentile(data[:, i], 75)
return median, perc_25, perc_75
def plot_data(ax, min_gen, max_gen, use_y_labels, use_legend):
# now all plot function should be applied to ax
ax.spines['top'].set_visible(False)
ax.spines['right'].set_visible(False)
ax.spines['left'].set_visible(False)
ax.get_xaxis().tick_bottom()
ax.get_yaxis().tick_left()
ax.tick_params(axis='x', direction='out')
ax.tick_params(axis='y', length=0)
# offset the spines
for spine in ax.spines.values():
spine.set_position(('outward', 5))
ax.grid(axis='y', color="0.9", linestyle='-', linewidth=1)
# put the grid behind
ax.set_axisbelow(True)
ax.fill_between(x, perc_25_low_mut, perc_75_low_mut, alpha=0.25, linewidth=0, color=colors[0])
ax.fill_between(x, perc_25_high_mut, perc_75_high_mut, alpha=0.25, linewidth=0, color=colors[1])
ax.plot(x, med_low_mut, linewidth=2, color=colors[0])
ax.plot(x, med_high_mut, linewidth=2, linestyle='--', color=colors[1])
# change xlim to set_xlim
ax.set_xlim(min_gen, max_gen)
ax.set_ylim(-5000, 300)
# change xticks to set_xticks
ax.set_xticks(np.arange(min_gen, max_gen, 100))
if not use_y_labels:
ax.set_yticklabels([])
if use_legend:
legend = ax.legend(["Low mutation rate", "High Mutation rate"], loc=4)
frame = legend.get_frame()
frame.set_facecolor('1.0')
frame.set_edgecolor('1.0')
data_low_mut = load('data/low_mut')
data_high_mut = load('data/high_mut')
n_generations = data_low_mut.shape[1]
x = np.arange(0, n_generations)
med_low_mut, perc_25_low_mut, perc_75_low_mut = perc(data_low_mut)
med_high_mut, perc_25_high_mut, perc_75_high_mut = perc(data_high_mut)
fig = figure()
fig.subplots_adjust(left=0.09, right=0.99, top=0.99, wspace=0.1)
ax1 = fig.add_subplot(121)
ax2 = fig.add_subplot(122)
plot_data(ax1, 0, 500, True, True)
plot_data(ax2, 0, 110, False, False)
# labeling
fig.text(0.01, 0.98, "A", weight="bold", horizontalalignment='left', verticalalignment='center')
fig.text(0.54, 0.98, "B", weight="bold", horizontalalignment='left', verticalalignment='center')
fig.savefig('variance_subplot_ter.png')