# Enforce the above in a plot format dictionary corresponding to data
pltfmt = OrderedDict([])
pltfmt['color'] = ['red' for i in xrange(0, nentries_brendan)
] + ['blue' for i in xrange(0, nentries_cone)]
pltfmt['marker'] = ['*' for i in xrange(0, nentries_brendan)
] + ['D' for i in xrange(0, nentries_cone)]
pltfmt['linestyle'] = [
'None' for i in xrange(0, nentries_brendan + nentries_cone)
]
# Plot final variables vs. initial burned mass for all cases
for h in headers:
data = OrderedDict([])
plt.figure(1)
fig = plt.gcf()
csp = CustomScatterplot(fig)
data['y'] = data_fin[h]
data['x'] = data_ini['mass burned']
csp.splot(data, 'x', 'y', pltfmt)
fig = csp.getfig()
mlco = mlines.Line2D([], [],
color='red',
marker='*',
markersize=5,
label='CO (Brendan)')
mlcone = mlines.Line2D([], [],
color='blue',
marker='D',
markersize=5,
label='CONe Hybrid')
plt.legend(handles=[mlco, mlcone],
