Update plot_mag_color_1d_hist.py

Author: zaidikumail

diffhtwo/experimental/diagnostics/plot_mag_color_1d_hist.py

@@ -246,7 +246,7 @@ def plot_n_mag(
         ax[i].set_xlabel(dimension_labels[i])
 
     ax[0].set_ylabel("number density [Mpc$^{-3}$]")
-    ax[-1].legend(framealpha=0.5)
+    ax[-1].legend(framealpha=0.5, loc="upper left", bbox_to_anchor=(-1, 1.2), ncols=3)
 
     for i in range(0, n_bands):
         ax[i].set_ylim(1e-6, 5e-3)