Skip to content

Instantly share code, notes, and snippets.

@aht
Forked from mbostock/.block
Last active January 29, 2016 05:22
Show Gist options
  • Save aht/f5e66ff1d75865de4eeb to your computer and use it in GitHub Desktop.
Save aht/f5e66ff1d75865de4eeb to your computer and use it in GitHub Desktop.
Reingold–Tilford Tree

The tree layout implements the Reingold-Tilford algorithm for efficient, tidy arrangement of layered nodes. The depth of nodes is computed by distance from the root, leading to a ragged appearance. Radial orientations are also supported. Implementation based on work by Jeff Heer and Jason Davies using Buchheim et al.'s linear-time variant of the Reingold-Tilford algorithm. Data shows the Flare class hierarchy, also courtesy Jeff Heer.

Compare to this radial layout.

<!DOCTYPE html>
<meta charset="utf-8">
<style>
.node circle {
stroke-width: 1.5px;
}
.node {
font: 12px sans-serif;
}
.link {
fill: none;
stroke: #ccc;
stroke-width: 1.5px;
}
</style>
<body>
<script src="//d3js.org/d3.v3.min.js"></script>
<script>
var width = 1600,
height = 800;
var tree = d3.layout.tree()
.size([height, width - 320]);
var diagonal = d3.svg.diagonal()
.projection(function(d) { return [d.y, d.x]; });
var svg = d3.select("body").append("svg")
.attr("width", width)
.attr("height", height)
.append("g")
.attr("transform", "translate(40,0)");
var data = {"id":1,"isLeaf":false,"parent":-1,"feature":"","label":"All","probability":"NaN","size":"NaN","segmentid":"NaN","children":[{"id":2,"isLeaf":false,"parent":1,"feature":"NumberCustomerServiceCalls","label":"<= 3","probability":"NaN","size":"NaN","segmentid":"NaN","children":[{"id":3,"isLeaf":false,"parent":2,"feature":"InternationalPlan","label":"= no","probability":"NaN","size":"NaN","segmentid":"NaN","children":[{"id":4,"isLeaf":false,"parent":3,"feature":"TotalLocalMinutes","label":"> 600","probability":"NaN","size":"NaN","segmentid":"NaN","children":[{"id":5,"isLeaf":false,"parent":4,"feature":"TotalDayMinutes","label":"> 210","probability":"NaN","size":"NaN","segmentid":"NaN","children":[{"id":6,"isLeaf":true,"parent":5,"feature":"DataPlan","label":"= no","probability":0.135,"size":0.05,"segmentid":2,"children":[]},{"id":7,"isLeaf":true,"parent":5,"feature":"DataPlan","label":"= yes","probability":0.114,"size":0.08,"segmentid":5,"children":[]}]},{"id":8,"isLeaf":true,"parent":4,"feature":"TotalDayMinutes","label":"<= 210","probability":0.129,"size":0.08,"segmentid":3,"children":[]}]},{"id":9,"isLeaf":true,"parent":3,"feature":"TotalLocalMinutes","label":"<= 600","probability":0.06,"size":0.29,"segmentid":9,"children":[]}]},{"id":10,"isLeaf":false,"parent":2,"feature":"InternationalPlan","label":"= yes","probability":"NaN","size":"NaN","segmentid":"NaN","children":[{"id":11,"isLeaf":false,"parent":10,"feature":"TotalIntlMinutes","label":"<= 13","probability":"NaN","size":"NaN","segmentid":"NaN","children":[{"id":12,"isLeaf":true,"parent":11,"feature":"TotalIntlCalls","label":"<= 3","probability":0.081,"size":0.05,"segmentid":7,"children":[]},{"id":13,"isLeaf":true,"parent":11,"feature":"TotalIntlCalls","label":"> 3","probability":0.15,"size":0.04,"segmentid":1,"children":[]}]}]}]},{"id":14,"isLeaf":false,"parent":1,"feature":"NumberCustomerServiceCalls","label":"> 3","probability":"NaN","size":"NaN","segmentid":"NaN","children":[{"id":15,"isLeaf":false,"parent":14,"feature":"TotalLocalMinutes","label":"> 500","probability":"NaN","size":"NaN","segmentid":"NaN","children":[{"id":16,"isLeaf":true,"parent":15,"feature":"DataPlan","label":"= yes","probability":0.12,"size":0.08,"segmentid":4,"children":[]},{"id":17,"isLeaf":true,"parent":15,"feature":"DataPlan","label":"= no","probability":0.09,"size":0.06,"segmentid":6,"children":[]}]},{"id":18,"isLeaf":true,"parent":14,"feature":"TotalLocalMinutes","label":"<= 500","probability":0.075,"size":0.19,"segmentid":8,"children":[]}]}]}
var nodes = tree.nodes(data),
links = tree.links(nodes);
var link = svg.selectAll("path.link")
.data(links)
.enter().append("path")
.attr("class", "link")
.attr("d", diagonal);
var node = svg.selectAll("g.node")
.data(nodes)
.enter().append("g")
.attr("class", "node")
.attr("transform", function(d) { return "translate(" + d.y + "," + d.x + ")"; })
var color = d3.scale.linear()
.domain([0, 0.15])
.range(["white", "red"]);
node.append("circle")
.attr("r", function(d) { return d.isLeaf ? 150 * d.size : 4.5; })
.attr("fill", function(d) { return d.isLeaf ? color(d.probability) : "" });
node.append("text")
.attr("dx", -16)
.attr("dy", 3)
.attr("text-anchor", "end")
.text(function(d) { return d.feature + " " + d.label; });
node.append("text")
.attr("dx", 32)
.attr("dy", 3)
.attr("font-weight", "bold")
.attr("text-anchor", "start")
.text(function(d) { return d.isLeaf ? "Segment "+d.segmentid + " (churn: " + d.probability*100 + "%, size: " + d.size*100 + "%)" : ""; });
d3.select(self.frameElement).style("height", height + "px");
</script>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment