summaryrefslogtreecommitdiff
path: root/finance/flow.py
diff options
context:
space:
mode:
Diffstat (limited to 'finance/flow.py')
-rw-r--r--finance/flow.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/finance/flow.py b/finance/flow.py
index 836c137..443966c 100644
--- a/finance/flow.py
+++ b/finance/flow.py
@@ -3,6 +3,8 @@ from datetime import datetime
from decimal import Decimal
from typing import Generator
+import matplotlib.pyplot
+
@dataclasses.dataclass(kw_only=True, frozen=True)
class Flow:
@@ -64,3 +66,15 @@ def simulate(
values.append(value)
return (dates, values)
+
+
+def display(data: tuple[list[datetime], list[Decimal]]) -> None:
+ matplotlib.pyplot.plot(
+ list(data[0]), # type: ignore
+ [float(i) for i in data[1]],
+ label="Money",
+ )
+ matplotlib.pyplot.xlabel("Time")
+ matplotlib.pyplot.legend()
+ matplotlib.pyplot.xticks(rotation=45)
+ matplotlib.pyplot.show()