From 33ebf9caeada2a046cd139ff49a6c3ce80542262 Mon Sep 17 00:00:00 2001
From: junos <junos.lukan@ijs.si>
Date: Fri, 6 Aug 2021 18:38:21 +0200
Subject: [PATCH] Add demonstrations of new functions.

---
 exploration/expl_communication.py | 31 ++++++++++++++++++++++++++++---
 1 file changed, 28 insertions(+), 3 deletions(-)

diff --git a/exploration/expl_communication.py b/exploration/expl_communication.py
index 6a6ebc8..0c464c0 100644
--- a/exploration/expl_communication.py
+++ b/exploration/expl_communication.py
@@ -14,6 +14,7 @@
 # ---
 
 # %%
+# %matplotlib inline
 import os
 import sys
 
@@ -39,9 +40,6 @@ print(df_calls)
 # %%
 count_comms(df_calls)
 
-# %%
-enumerate_contacts(df_calls)
-
 # %%
 df_sms = get_sms_data(["nokia_0000003"])
 count_comms(df_sms)
@@ -141,3 +139,30 @@ sms_number = pd.wide_to_long(
 sns.displot(
     sms_number, x="no", hue="message_type", binwidth=5, element="step", height=8
 )
+
+# %% [markdown]
+# # Communication features
+
+# %%
+df_calls_enumerated = enumerate_contacts(df_calls)
+display(df_calls_enumerated)
+
+# %%
+df_calls_contact_features = contact_features(df_calls_enumerated)
+display(df_calls_contact_features)
+
+# %%
+df_sms_enumerated = enumerate_contacts(df_sms)
+df_sms_contact_features = contact_features(df_sms_enumerated)
+display(df_sms_contact_features)
+
+# %%
+display(count_comms(df_calls))
+
+# %%
+display(count_comms(df_sms))
+
+# %%
+display(calls_sms_features(df_calls, df_sms))
+
+# %%