summaryrefslogtreecommitdiff
path: root/finance/visualize.py
diff options
context:
space:
mode:
Diffstat (limited to 'finance/visualize.py')
-rw-r--r--finance/visualize.py19
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()