# 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],