From 13489fe2b0927eecc2e0c325f195a691c3116d66 Mon Sep 17 00:00:00 2001 From: MstrPikachu <31784486+MstrPikachu@users.noreply.github.com> Date: Sun, 12 Dec 2021 00:48:27 -0500 Subject: [PATCH] Add gridlines --- src/process/twitter_visualization.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/process/twitter_visualization.py b/src/process/twitter_visualization.py index 692469d..4192a4b 100644 --- a/src/process/twitter_visualization.py +++ b/src/process/twitter_visualization.py @@ -332,6 +332,9 @@ def graph_histogram(x: list[float], path: str, title: str, freq: bool, clear_out for spine in ax.spines.values(): spine.set_edgecolor(border_color) + # Grid + ax.grid(visible=True, axis='both') + # Save fig.savefig(os.path.join(REPORT_DIR, path)) fig.clf() @@ -424,6 +427,9 @@ def graph_line_plot(x: list[datetime], y: Union[list[float], list[list[float]]], for spine in ax.spines.values(): spine.set_edgecolor(border_color) + # Grid + ax.grid(visible=True, axis='both') + # Save path = Path(os.path.join(REPORT_DIR, path)) path.parent.mkdir(parents=True, exist_ok=True)