Added theme and labels back into python scripts

python_scripts/phase_transition.py

@@ -4,6 +4,20 @@ from pathlib import Path
 import matplotlib.pyplot as plt
 import numpy as np
 from scipy.stats import linregress
+import seaborn as sns
+
+sns.set_theme()
+params = {
+    "font.family": "Serif",
+    "font.serif": "Roman",
+    "text.usetex": True,
+    "axes.titlesize": "large",
+    "axes.labelsize": "large",
+    "xtick.labelsize": "large",
+    "ytick.labelsize": "large",
+    "legend.fontsize": "medium",
+}
+plt.rcParams.update(params)
 
 
 def plot_phase_transition_alt(indir, outdir):
@@ -65,20 +79,40 @@ def plot_phase_transition_alt(indir, outdir):
     x = np.linspace(0, 1 / 20, 1001)
     regression = linregress(inv_L, Tc)
     f = lambda x: regression[0] * x + regression[1]
+    stats = (
+        f"$\\beta_{0}$ = {regression[1]:.4f}\n"
+        f"$\\beta_{1}$ = {regression[0]:.4f}"
+    )
+    bbox = dict(boxstyle="round", pad=0.3, fc="white", ec="white", alpha=0.5)
+    ax5.text(
+        0.6, 0.6, stats, bbox=bbox, transform=ax1.transAxes, ha="right", va="center"
+    )
     ax5.scatter(inv_L, Tc)
-    ax5.plot(x, f(x), label=f"m = {regression[0]}, i = {regression[1]}")
+    ax5.plot(x, f(x))
 
-    figure1.legend()
-    figure2.legend()
-    figure3.legend()
-    figure4.legend()
-    figure5.legend()
+    ax1.set_xlabel(r"T $(J/k_{B})$")
+    ax1.set_ylabel(r"$\langle \epsilon \rangle$ $(J)$")
+    ax1.legend(title="Lattice size", loc="upper right")
 
-    figure1.savefig(Path(outdir, "energy.pdf"))
-    figure2.savefig(Path(outdir, "magnetization.pdf"))
-    figure3.savefig(Path(outdir, "heat_capacity.pdf"))
-    figure4.savefig(Path(outdir, "susceptibility.pdf"))
-    figure5.savefig(Path(outdir, "linreg.pdf"))
+    ax2.set_xlabel(r"T $(J/k_{B})$")
+    ax2.set_ylabel(r"$\langle |m| \rangle$ $(unitless)$")
+    ax2.legend(title="Lattice size", loc="upper right")
+
+    ax3.set_xlabel(r"T $(J/k_{B})$")
+    ax3.set_ylabel(r"$C_{V}$ $(k_{B})$")
+    ax3.legend(title="Lattice size", loc="upper right")
+
+    ax4.set_xlabel(r"T $(J/k_{B})$")
+    ax4.set_ylabel(r"$\chi$ $(1/J)$")
+    ax4.legend(title="Lattice size", loc="upper right")
+
+    # ax5.legend()
+
+    figure1.savefig(Path(outdir, "energy.pdf"), bbox_inches="tight")
+    figure2.savefig(Path(outdir, "magnetization.pdf"), bbox_inches="tight")
+    figure3.savefig(Path(outdir, "heat_capacity.pdf"), bbox_inches="tight")
+    figure4.savefig(Path(outdir, "susceptibility.pdf"), bbox_inches="tight")
+    figure5.savefig(Path(outdir, "linreg.pdf"), bbox_inches="tight")
 
     plt.close(figure1)
     plt.close(figure2)
@@ -99,11 +133,11 @@ def plot_phase_transition(indir, outdir):
         "size_100.txt",
     ]
     labels = [
-        "L = 20",
-        "L = 40",
-        "L = 60",
-        "L = 80",
-        "L = 100",
+        "20",
+        "40",
+        "60",
+        "80",
+        "100",
     ]
 
     figure1, ax1 = plt.subplots()
@@ -151,20 +185,40 @@ def plot_phase_transition(indir, outdir):
     x = np.linspace(0, 1 / 20, 1001)
     regression = linregress(inv_L, Tc)
     f = lambda x: regression[0] * x + regression[1]
+    stats = (
+        f"$\\beta_{0}$ = {regression[1]:.4f}\n"
+        f"$\\beta_{1}$ = {regression[0]:.4f}"
+    )
+    bbox = dict(boxstyle="round", pad=0.3, fc="white", ec="white", alpha=0.5)
+    ax5.text(
+        0.6, 0.6, stats, bbox=bbox, transform=ax1.transAxes, ha="right", va="center"
+    )
     ax5.scatter(inv_L, Tc)
-    ax5.plot(x, f(x), label=f"m = {regression[0]}, i = {regression[1]}")
+    ax5.plot(x, f(x))
 
-    figure1.legend()
-    figure2.legend()
-    figure3.legend()
-    figure4.legend()
-    figure5.legend()
+    ax1.set_xlabel(r"T $(J/k_{B})$")
+    ax1.set_ylabel(r"$\langle \epsilon \rangle$ $(J)$")
+    ax1.legend(title="Lattice size", loc="upper right")
 
-    figure1.savefig(Path(outdir, "energy.pdf"))
-    figure2.savefig(Path(outdir, "magnetization.pdf"))
-    figure3.savefig(Path(outdir, "heat_capacity.pdf"))
-    figure4.savefig(Path(outdir, "susceptibility.pdf"))
-    figure5.savefig(Path(outdir, "linreg.pdf"))
+    ax2.set_xlabel(r"T $(J/k_{B})$")
+    ax2.set_ylabel(r"$\langle |m| \rangle$ $(unitless)$")
+    ax2.legend(title="Lattice size", loc="upper right")
+
+    ax3.set_xlabel(r"T $(J/k_{B})$")
+    ax3.set_ylabel(r"$C_{V}$ $(k_{B})$")
+    ax3.legend(title="Lattice size", loc="upper right")
+
+    ax4.set_xlabel(r"T $(J/k_{B})$")
+    ax4.set_ylabel(r"$\chi$ $(1/J)$")
+    ax4.legend(title="Lattice size", loc="upper right")
+
+    # ax5.legend()
+
+    figure1.savefig(Path(outdir, "energy.pdf"), bbox_inches="tight")
+    figure2.savefig(Path(outdir, "magnetization.pdf"), bbox_inches="tight")
+    figure3.savefig(Path(outdir, "heat_capacity.pdf"), bbox_inches="tight")
+    figure4.savefig(Path(outdir, "susceptibility.pdf"), bbox_inches="tight")
+    figure5.savefig(Path(outdir, "linreg.pdf"), bbox_inches="tight")
 
     plt.close(figure1)
     plt.close(figure2)