mirror of
https://github.com/NetherlandsForensicInstitute/hansken-python-workshop.git
synced 2026-08-20 04:47:33 +00:00
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
# %% [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()
|