diff --git a/src/plot.py b/src/plot.py index 511ba8c..adacd80 100644 --- a/src/plot.py +++ b/src/plot.py @@ -27,8 +27,7 @@ def plot_transformations(save: bool=False) -> None: def plot_eigenvectors(N: int, save: bool=False) -> None: # Load data based on matrix size - name = "eigenvector_" + str(N) - path = "../latex/output/" + name + ".csv" + path = f"../latex/output/eigenvector_{N}.csv" eigvec = pd.read_csv(path, header=0) fig, ax = plt.subplots() @@ -41,7 +40,7 @@ def plot_eigenvectors(N: int, save: bool=False) -> None: # Save to file if save is True: - fig.savefig("../latex/images/" + name + ".pdf") + fig.savefig(f"../latex/images/eigenvector_{N}.pdf")