FYS3150/Project-2
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use f-strings

Browse Source
This commit is contained in:
Janita Willumsen 2023-09-23 16:22:56 +02:00
parent 97e6b35112
commit 3c9a84598d
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/plot.py
View File

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