initial commit

unique_values_treemap_chatmessage.py

@@ -0,0 +1,41 @@
+# %% [markdown]
+## Plot the distribution of senders of chat messages
+### Setup Hansken connection
+# %% [python]
+import sys
+import squarify
+from types import SimpleNamespace
+import matplotlib.pyplot as plt
+
+from hansken.connect import connect_project
+
+in_browser = 'js' in sys.modules
+hansken_host = ''
+context = connect_project(endpoint=f'http://{hansken_host}:9091/gatekeeper/',
+                          project='5ee273fd-0978-4a0a-b8b0-2af2f8479214',
+                          keystore=f'http://{hansken_host}:9091/keystore/',
+                          # Authentication is faked if we run in the browser,
+                          # because an authenticated session should already be present
+                          auth=SimpleNamespace() if in_browser else None,
+                          interactive=True)
+
+# %% [markdown]
+### Retrieve all senders
+# The `unique_values` function returns all values for a given property within a project. In this case, we retrieve all values for `chatMessage.from`.
+
+# %% [python]
+sizes = []
+labels = []
+for sender in context.unique_values("chatMessage.from"):
+    sizes.append(sender['count'])
+    labels.append(sender['value'])
+
+# %% [markdown]
+### Use a treemap visualization to plot the distribution of senders.
+
+# %% [python]
+fig = plt.figure(figsize=(12,6))
+ax = fig.add_subplot(111)
+squarify.plot(sizes=sizes, label=labels, alpha=.6, ax=ax)
+plt.axis('off')
+plt.show()