From c692a01f72ccb923618cee6c6de73734726440f9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lu=C3=ADs=20Murta?= Date: Sat, 24 Apr 2021 00:27:37 +0100 Subject: [PATCH] Graph now saves the images and increases its width --- pfbudget/graph.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/pfbudget/graph.py b/pfbudget/graph.py index ce2443d..98a71b2 100644 --- a/pfbudget/graph.py +++ b/pfbudget/graph.py @@ -69,6 +69,7 @@ def monthly(state, start, end): ) ) + plt.figure(figsize=(30, 10)) plt.plot(daterange(start, end, "month"), income, label="Income") plt.stackplot( daterange(start, end, "month"), @@ -79,7 +80,8 @@ def monthly(state, start, end): labels=["Fixed", "Required", "Health", "Discretionary"], ) plt.legend(loc="upper left") - plt.show() + plt.tight_layout() + plt.savefig("graph.png") def discrete(state, start, end): @@ -138,6 +140,7 @@ def discrete(state, start, end): # transposing discretionary discretionary = list(map(list, zip(*discretionary))) + plt.figure(figsize=(30, 10)) plt.plot(daterange(start, end, "month"), income, label="Income") plt.stackplot( daterange(start, end, "month"), @@ -149,7 +152,8 @@ def discrete(state, start, end): ) plt.legend(loc="upper left") plt.grid() - plt.show() + plt.tight_layout() + plt.savefig("graph.png") def average(state, start, end): @@ -219,6 +223,7 @@ def average(state, start, end): print(avg_discretionary) + plt.figure(figsize=(30, 10)) plt.plot(daterange(start, end, "month"), l_avg_income, label=f"Income {avg_income}") plt.stackplot( daterange(start, end, "month"), @@ -237,4 +242,8 @@ def average(state, start, end): ], ) plt.legend(bbox_to_anchor=(1, 1), loc="upper left") - plt.show() + plt.tight_layout() + plt.savefig( + "graph.png", + dpi=600, + )