From 7c2ea6f5bed8db81ee83f57e8d42954d7d1862c6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment=20Walter?= <clement0walter@gmail.com>
Date: Wed, 6 Sep 2023 07:37:29 +0200
Subject: [PATCH] Add monthly plot

---
 packages/starksheet-cairo/notebooks/sheets.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/packages/starksheet-cairo/notebooks/sheets.py b/packages/starksheet-cairo/notebooks/sheets.py
index 0f968fe..bba9ab5 100644
--- a/packages/starksheet-cairo/notebooks/sheets.py
+++ b/packages/starksheet-cairo/notebooks/sheets.py
@@ -134,6 +134,21 @@
 plt.tight_layout()
 plt.savefig("daily_sheets.png")
 
+# %% Plot monthly
+counts = (
+    calls.groupby(["contract_identifier", pd.Grouper(key="timestamp", freq="M")])
+    .size()
+    .unstack("contract_identifier", fill_value=0)
+    .fillna(0)
+    .astype(int)
+)
+ax = counts.plot(kind="bar", stacked=True)
+x_labels = [d.date().strftime("%Y-%m") for d in counts.index]
+ax.set_xticklabels(x_labels)
+ax.grid(axis="y", linestyle="--", color="grey")
+plt.tight_layout()
+plt.savefig("monthly_sheets.png")
+
 # %% Plot hourly sheet creation
 plt.clf()
 ax = (