diff options
Diffstat (limited to 'finance')
-rw-r--r-- | finance/flow.py | 14 |
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() |