Rahka commited on
Commit
e8fb6a3
·
1 Parent(s): 4304f13

add taxonomy tree map with 200 labels

Browse files
Files changed (1) hide show
  1. app.py +38 -1
app.py CHANGED
@@ -1,3 +1,40 @@
1
  import streamlit as st
 
2
 
3
- st.title('Musterdatenkatalog')
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import streamlit as st
2
+ import json
3
 
4
+ with open("taxonomy_v3", "rb") as fp:
5
+ taxonomy = json.load(fp)
6
+
7
+ taxonomy = taxonomy[:200]
8
+
9
+ names = [""]
10
+ parents = ["root"]
11
+
12
+ taxonomy_group_label_mapper = {el["group"]: [] for el in taxonomy}
13
+
14
+ for el in taxonomy:
15
+ taxonomy_group_label_mapper[el["group"]].append(el["label"])
16
+
17
+ for group, labels in taxonomy_group_label_mapper.items():
18
+ names.append(group)
19
+ parents.append("root")
20
+ for label in labels:
21
+ names.append(label)
22
+ parents.append(group)
23
+ if "-" in label:
24
+ names.append(label.split("-")[0])
25
+ parents.append(label)
26
+ names.append(label.split("-")[1])
27
+ parents.append(label)
28
+
29
+ fig = px.treemap(
30
+ names=names,
31
+ parents=parents,
32
+ )
33
+ fig.update_traces(root_color="lightgrey")
34
+ fig.update_layout(margin=dict(t=100, l=50, r=50, b=50))
35
+
36
+
37
+ st.title('Musterdatenkatalog')
38
+
39
+
40
+ st.plotly_chart(fig)