Skip to content

Instantly share code, notes, and snippets.

@silviaclaire

silviaclaire/app.py

Last active August 11, 2023 19:37
Show Gist options
  • Star 5 You must be signed in to star a gist
  • Fork 3 You must be signed in to fork a gist
  • Save silviaclaire/554938c808f5303601ec9cc0cefdd866 to your computer and use it in GitHub Desktop.
Save silviaclaire/554938c808f5303601ec9cc0cefdd866 to your computer and use it in GitHub Desktop.
Download ZIP
Collapsible Tree Diagram using Flask and d3.js
Raw
app.py
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()))
Raw
data.csv
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
Raw
index.html
<!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