Start implementing plot functionality

This commit is contained in:
Janita Willumsen 2023-09-22 08:38:30 +02:00
parent 162e884d47
commit f85e41460b

45
src/plot.py Normal file
View File

@ -0,0 +1,45 @@
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
sns.set_theme()
sns.dark_palette("seagreen")
"""Write to file (for dense matrix):
Line give size of matrix (column give number of transformations until convergence)
N 1 (2 3 4 5)
3
6
9
12
15
"""
def read_data(filename: str) -> tuple:
N = []
T = []
with open(filename, "r") as f:
lines = f.readlines()
for line in lines:
n_i, t_i = line.strip().split(",")
N.append(float(n_i))
T.append(float(t_i))
return (N, T)
def plot_similarity_transformations(N: np.ndarray, T: np.ndarray) -> None:
fig, ax = plt.subplots()
ax.plot(N, T, label='Transformations')
ax.set_xlabel('N')
ax.set_ylabel('Similarity transformations')
ax.set_xlim(xmin=N[0], xmax=N[-1])
fig.savefig("similarity_transformation.pdf")
def plot_eigenvectors():
pass
if __name__ == '__main__':
pass