Last active
August 11, 2023 19:37
-
-
Save silviaclaire/554938c808f5303601ec9cc0cefdd866 to your computer and use it in GitHub Desktop.
Collapsible Tree Diagram using Flask and d3.js
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import csv | |
import json | |
from flask import Flask, render_template | |
# reference: https://stackoverflow.com/questions/29631711/python-to-parent-child-json | |
# create a helper class for each tree node | |
class Node(object): | |
# generate new node | |
def __init__(self, cluster): | |
self.cluster = cluster | |
self.children = [] | |
# append a child node to parent (self) | |
def child(self, child_cluster): | |
# check if child already exists in parent | |
child_found = [c for c in self.children if c.cluster == child_cluster] | |
if not child_found: | |
# if it is a new child, create new node for this child | |
_child = Node(child_cluster) | |
# append new child node to parent's children list | |
self.children.append(_child) | |
else: | |
# if the same, save this child | |
_child = child_found[0] | |
# return child object for later add | |
return _child | |
# convert the whole object to dict | |
def as_dict(self): | |
res = {'cluster': self.cluster} | |
res['children'] = [c.as_dict() for c in self.children] | |
return res | |
# drive function | |
app = Flask(__name__) | |
@app.route('/') | |
def create_tree(): | |
root = Node('data') | |
cluster_levels = 3 | |
with open('data.csv', 'r') as f: | |
reader = csv.reader(f) | |
# skip header row | |
reader.next() | |
# scan data by row | |
for row in reader: | |
parent = root | |
# cluster info: from column 1 to 3 | |
for level in range(1, (cluster_levels + 1)): | |
parent = parent.child(row[level]) | |
# print(json.dumps(root.as_dict(), indent=4)) | |
return render_template('index.html', data = json.dumps(root.as_dict())) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
data-id | cluster-id | cluster-id | cluster-id | |
---|---|---|---|---|
5 | 1 | 1-1 | 1-1-1 | |
6 | 1 | 1-1 | 1-1-2 | |
45 | 1 | 1-2 | 1-2-1 | |
75 | 1 | 1-3 | 1-3-1 | |
76 | 1 | 1-3 | 1-3-2 | |
676 | 2 | 2-1 | 2-1-1 | |
2345 | 2 | 2-2 | 2-2-1 | |
67 | 3 | 3-1 | 3-1-1 | |
734 | 3 | 3-2 | 3-2-1 | |
734 | 3 | 3-2 | 3-2-2 | |
34 | 3 | 3-3 | 3-3-1 | |
121 | 4 | 4-1 | 4-1-1 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
<!DOCTYPE html> | |
<meta charset="utf-8"> | |
<style> | |
.node { | |
cursor: pointer; | |
} | |
.node circle { | |
fill: #fff; | |
stroke: steelblue; | |
stroke-width: 1.5px; | |
} | |
.node text { | |
font: 10px sans-serif; | |
} | |
.link { | |
fill: none; | |
stroke: #ccc; | |
stroke-width: 1.5px; | |
} | |
</style> | |
<body> | |
<script src="//d3js.org/d3.v3.min.js"></script> | |
<script> | |
// reference: https://bl.ocks.org/mbostock/4339083 | |
treeDiagram({{ data|safe }}) | |
function treeDiagram(root) { | |
var margin = {top: 20, right: 120, bottom: 20, left: 120}, | |
width = 960 - margin.right - margin.left, | |
height = 800 - margin.top - margin.bottom; | |
var i = 0, | |
duration = 750, | |
root; | |
var tree = d3.layout.tree() | |
.size([height, width]); | |
var diagonal = d3.svg.diagonal() | |
.projection(function(d) { return [d.y, d.x]; }); | |
var svg = d3.select("body").append("svg") | |
.attr("width", width + margin.right + margin.left) | |
.attr("height", height + margin.top + margin.bottom) | |
.append("g") | |
.attr("transform", "translate(" + margin.left + "," + margin.top + ")"); | |
console.log(root); | |
root.x0 = height / 2; | |
root.y0 = 0; | |
update(root); | |
function update(source) { | |
// Compute the new tree layout. | |
var nodes = tree.nodes(root).reverse(), | |
links = tree.links(nodes); | |
// Normalize for fixed-depth. | |
nodes.forEach(function(d) { d.y = d.depth * 180; }); | |
// Update the nodes… | |
var node = svg.selectAll("g.node") | |
.data(nodes, function(d) { return d.id || (d.id = ++i); }); | |
// Enter any new nodes at the parent's previous position. | |
var nodeEnter = node.enter().append("g") | |
.attr("class", "node") | |
.attr("transform", function(d) { return "translate(" + source.y0 + "," + source.x0 + ")"; }) | |
nodeEnter.append("circle") | |
.attr("r", 1e-6) | |
.style("fill", function(d) { return d._children ? "lightsteelblue" : "#fff"; }); | |
nodeEnter.append("text") | |
.attr("x", function(d) { return d.children || d._children ? -10 : 10; }) | |
.attr("dy", ".35em") | |
.attr("text-anchor", function(d) { return d.children || d._children ? "end" : "start"; }) | |
.text(function(d) { return d.cluster; }) | |
.style("fill-opacity", 1e-6); | |
// Transition nodes to their new position. | |
var nodeUpdate = node.transition() | |
.duration(duration) | |
.attr("transform", function(d) { return "translate(" + d.y + "," + d.x + ")"; }); | |
nodeUpdate.select("circle") | |
.attr("r", 4.5) | |
.style("fill", function(d) { return d._children ? "lightsteelblue" : "#fff"; }); | |
nodeUpdate.select("text") | |
.style("fill-opacity", 1); | |
// Transition exiting nodes to the parent's new position. | |
var nodeExit = node.exit().transition() | |
.duration(duration) | |
.attr("transform", function(d) { return "translate(" + source.y + "," + source.x + ")"; }) | |
.remove(); | |
nodeExit.select("circle") | |
.attr("r", 1e-6); | |
nodeExit.select("text") | |
.style("fill-opacity", 1e-6); | |
// Update the links… | |
var link = svg.selectAll("path.link") | |
.data(links, function(d) { return d.target.id; }); | |
// Enter any new links at the parent's previous position. | |
link.enter().insert("path", "g") | |
.attr("class", "link") | |
.attr("d", function(d) { | |
var o = {x: source.x0, y: source.y0}; | |
return diagonal({source: o, target: o}); | |
}); | |
// Transition links to their new position. | |
link.transition() | |
.duration(duration) | |
.attr("d", diagonal); | |
// Transition exiting nodes to the parent's new position. | |
link.exit().transition() | |
.duration(duration) | |
.attr("d", function(d) { | |
var o = {x: source.x, y: source.y}; | |
return diagonal({source: o, target: o}); | |
}) | |
.remove(); | |
// Stash the old positions for transition. | |
nodes.forEach(function(d) { | |
d.x0 = d.x; | |
d.y0 = d.y; | |
}); | |
} | |
} | |
</script> |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment