From 3d7cff4ae7edc74f0e69c41929b19aa72be55d52 Mon Sep 17 00:00:00 2001 From: niklasfranz <nf.app@icloud.com> Date: Tue, 11 Jul 2023 03:44:38 +0200 Subject: [PATCH] feat: added plot_scatterplot function --- src/Plotter.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/Plotter.py b/src/Plotter.py index c16adfe..8e43acc 100644 --- a/src/Plotter.py +++ b/src/Plotter.py @@ -43,4 +43,9 @@ class Plotter: weights=anx_score_weights, bins = bins, alpha=0.5, - ) \ No newline at end of file + ) + + def plot_scatterplot(self, target1, target2, styling_params = {}): + fig, ax = plt.subplots() + self.customize_plot(fig, ax, styling_params) + ax.scatter(self.df[target1], self.df[target2]) \ No newline at end of file -- GitLab