fig, ax = pyplot.subplots(figsize=(12, 10))
# use gridspec to allow for one plot to be larger than the others
gs = gridspec.GridSpec(3, 6, height_ratios=[2, 0.5, 0.5])
ax = []
ax.append(pyplot.subplot(gs[0, 0:3]))
ax.append(pyplot.subplot(gs[0, 3:6]))
ax.append(pyplot.subplot(gs[1, 0:2]))
ax.append(pyplot.subplot(gs[1, 2:4]))
ax.append(pyplot.subplot(gs[1, 4:6]))
ax.append(pyplot.subplot(gs[2, 0:2]))
ax.append(pyplot.subplot(gs[2, 2:4]))
ax.append(pyplot.subplot(gs[2, 4:6]))
plot_indiv_comp(ax[0], ["1e0", "1e1"], ["090"], "Total", "dirty_nz", r"$n_{z}$", fontsize=fontsize)
leg = ax[0].legend(fontsize=0.95 * fontsize, loc="lower left")
leg.get_frame().set_linewidth(2)
ax[0].set_title("Global SED")
ax[0].set_xlim(2, 100)
plot_converge_slice(ax[1], ["1e0", "1e1"], ["035.11", "151.99"], "090", "dirty_nz", r"$n_{z}$", fontsize=fontsize)
ax[1].yaxis.tick_right()
ax[1].yaxis.set_label_position("right")
leg = ax[1].legend(fontsize=0.95 * fontsize, loc="lower left")
leg.get_frame().set_linewidth(2)
ax[1].set_title(r"Y Image Slice, $\theta = 090^\circ$")
ax[1].set_xlim(2, 100)
modname = "dirty_nbinz"
cvals = ["2", "5", "10", "20", "50", "100"]
ax.append(plt.subplot(gs[1,5]))
ax.append(plt.subplot(gs[2,5]))
ax.append(plt.subplot(gs[0,6]))
ax.append(plt.subplot(gs[1,6]))
ax.append(plt.subplot(gs[2,6]))
ax.append(plt.subplot(gs[0,7]))
ax.append(plt.subplot(gs[1,7]))
ax.append(plt.subplot(gs[2,7]))
ax.append(plt.subplot(gs[0,8]))
ax.append(plt.subplot(gs[1,8]))
ax.append(plt.subplot(gs[2,8]))
plot_indiv_comp(ax[0], ['1e0','1e1'], ['000','090','180'], 'Total',
'dirty_nphot', r'$N$', fontsize=fontsize)
# Create two custom legends (more compact)
# taus
leg1 = ax[0].legend([plt.Line2D((0,1),(0,0), color='k', linestyle='-'),
plt.Line2D((0,1),(0,0), color='k', linestyle='--')],
[r'$\tau_z = 1e0$',
r'$\tau_z = 1e1$'],
fontsize=fontsize,
loc='lower left')
leg1.get_frame().set_linewidth(2)
# Add the legend manually to the current Axes.
ax[0].add_artist(leg1)
fig, ax = pyplot.subplots(figsize=(12,8))
# use gridspec to allow for one plot to be larger than the others
gs = gridspec.GridSpec(2, 6, height_ratios=[2,1])
ax = []
ax.append(pyplot.subplot(gs[0,0:3]))
ax.append(pyplot.subplot(gs[0,3:6]))
ax.append(pyplot.subplot(gs[1,0]))
ax.append(pyplot.subplot(gs[1,1]))
ax.append(pyplot.subplot(gs[1,2]))
ax.append(pyplot.subplot(gs[1,3]))
ax.append(pyplot.subplot(gs[1,4]))
ax.append(pyplot.subplot(gs[1,5]))
plot_indiv_comp(ax[0], ['1e0','1e1'], ['000','180'], 'Total',
'dirty_nxy', r'$n_{xy}$', fontsize=fontsize)
leg = ax[0].legend(fontsize=0.95*fontsize, loc='lower left')
leg.get_frame().set_linewidth(2)
ax[0].set_title('Global SED')
ax[0].set_xlim(2,100)
plot_converge_slice(ax[1], ['1e0','1e1'], ['035.11','151.99'], '000',
'dirty_nxy', r'$n_{xy}$', fontsize=fontsize)
ax[1].yaxis.tick_right()
ax[1].yaxis.set_label_position("right")
leg = ax[1].legend(fontsize=0.95*fontsize, loc='lower left')
leg.get_frame().set_linewidth(2)
ax[1].set_title(r'Y Image Slice, $\theta = 000^\circ$')
ax[1].set_xlim(2,100)
modname = 'dirty_nbinxy'