From 9faaac7e25d2efdca36dcb67f96960d3ab72ab82 Mon Sep 17 00:00:00 2001 From: xengineering Date: Sun, 8 Sep 2024 11:03:45 +0200 Subject: Implement finance.flow.display() --- finance/flow.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'finance') 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() -- cgit v1.2.3-70-g09d2