diff options
Diffstat (limited to 'finance/visualize.py')
-rw-r--r-- | finance/visualize.py | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/finance/visualize.py b/finance/visualize.py index e6d609a..c7a7c60 100644 --- a/finance/visualize.py +++ b/finance/visualize.py @@ -4,13 +4,24 @@ from decimal import Decimal import matplotlib.pyplot -def display(data: tuple[list[datetime], list[Decimal]]) -> None: +def display( + simulated: tuple[list[datetime], list[Decimal]], + measured: list[tuple[datetime, Decimal]], +) -> None: matplotlib.pyplot.plot( - list(data[0]), # type: ignore - [float(i) for i in data[1]], - label="Money", + list(simulated[0]), # type: ignore + [float(i) for i in simulated[1]], + label="Simulated", + ) + matplotlib.pyplot.plot( + [i[0] for i in measured], # type: ignore + [float(i[1]) for i in measured], + label="Measured", + marker="x", + linestyle="None", ) matplotlib.pyplot.xlabel("Time") + matplotlib.pyplot.ylabel("Money") matplotlib.pyplot.legend() matplotlib.pyplot.xticks(rotation=45) matplotlib.pyplot.show() |