From 2fde5922560f5261fd8c938f9a3ad967527fb8cf Mon Sep 17 00:00:00 2001 From: Janita Willumsen Date: Mon, 25 Sep 2023 13:11:04 +0200 Subject: [PATCH] Include analytic vector in plot --- src/plot.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/src/plot.py b/src/plot.py index adacd80..4daaebc 100644 --- a/src/plot.py +++ b/src/plot.py @@ -31,12 +31,15 @@ def plot_eigenvectors(N: int, save: bool=False) -> None: eigvec = pd.read_csv(path, header=0) fig, ax = plt.subplots() - ax.plot(eigvec['x'], eigvec['Vector 1'], label='Vector 1') - ax.plot(eigvec['x'], eigvec['Vector 2'], label='Vector 2') - ax.plot(eigvec['x'], eigvec['Vector 3'], label='Vector 3') + ax.plot(eigvec['x'], eigvec['Vector 1'], label=r'$\vec{v}_{1}$') + ax.plot(eigvec['x'], eigvec['Vector 2'], label=r'$\vec{v}_{2}$') + ax.plot(eigvec['x'], eigvec['Vector 3'], label=r'$\vec{v}_{3}$') + ax.plot(eigvec['x'], eigvec['Analytic 1'], '--', label=r'$\vec{v}^{(1)}$') + ax.plot(eigvec['x'], eigvec['Analytic 2'], '--', label=r'$\vec{v}^{(2)}$') + ax.plot(eigvec['x'], eigvec['Analytic 3'], '--', label=r'$\vec{v}^{(3)}$') ax.set_xlabel(r'Element $\hat{x}_{i}$') - ax.set_ylabel(r'Value of element $v_{i}$') - ax.legend() + ax.set_ylabel(r'Element $v_{i}$') + ax.legend(loc='upper left') # Save to file if save is True: @@ -46,7 +49,7 @@ def plot_eigenvectors(N: int, save: bool=False) -> None: if __name__ == '__main__': - plot_transformations(True) + # plot_transformations(True) plot_eigenvectors(6, True) plot_eigenvectors(100, True) # plt.show() \ No newline at end of file