Skip to content

Instantly share code, notes, and snippets.

@elktamer
Last active February 28, 2018 20:23
Show Gist options
  • Save elktamer/b09b7f8e0dc3240d8ea0b03626891370 to your computer and use it in GitHub Desktop.
Save elktamer/b09b7f8e0dc3240d8ea0b03626891370 to your computer and use it in GitHub Desktop.
Federal Budget 2018
source target value
Personal income tax Income taxes 143.2
Corporate income tax Income taxes 42.5
Non-resident income tax Income taxes 6.6
Income taxes Tax Revenue 192.4
Goods and Services Tax Excise taxes and duties 33.7
Customs import duties Excise taxes and duties 5.4
Other excise taxes/duties Excise taxes and duties 11.6
Excise taxes and duties Tax Revenue 50.6
Tax Revenue Revenue 243
EI premiums Revenue 22.3
Crown corporations Other Revenue 10.1
Other programs Other Revenue 14.7
Net foreign exchange Other Revenue 1.9
Other Revenue Revenue 26.7
Transfers to persons Elderly benefits 48.3
Transfers to persons EI benefits 21
Transfers to persons Childrens benefits 21.9
Expenses Transfers to persons 91.2
Transfers to governments Canada Health Transfer 36.1
Transfers to governments Canada Social Transfer 13.3
Transfers to governments Equalization 17.9
Transfers to governments Territorial Formula Financing 3.6
Transfers to governments Gas Tax Fund 2.1
Transfers to governments Other fiscal arrangements -4.3
Expenses Transfers to governments 68.7
Direct program expenses Transfer payments 42.7
Direct program expenses Capital amortization 5
Direct program expenses Operating expenses 83.2
Expenses Direct program expenses 130.9
Expenses Public debt charges 24.3
Revenue Expenses 292.1
Deficit Expenses 5.4
source target value
Personal income tax Income taxes 152.1
Corporate income tax Income taxes 43.6
Non-resident income tax Income taxes 6.9
Income taxes Tax Revenue 202.6
Goods and Services Tax Excise taxes and duties 35.1
Customs import duties Excise taxes and duties 4.9
Other excise taxes/duties Excise taxes and duties 11.7
Excise taxes and duties Tax Revenue 51.7
Tax Revenue Revenue 254.3
EI premiums Revenue 21.2
Crown corporations Other Revenue 11.4
Other programs Other Revenue 15.8
Net foreign exchange Other Revenue 1.9
Other Revenue Revenue 29.1
Transfers to persons Elderly benefits 51.1
Transfers to persons EI benefits 22
Transfers to persons Childrens benefits 23
Expenses Transfers to persons 96.1
Transfers to governments Canada Health Transfer 37.1
Transfers to governments Canada Social Transfer 13.7
Transfers to governments Equalization 18.3
Transfers to governments Territorial Formula Financing 3.7
Transfers to governments Gas Tax Fund 2.1
Transfers to governments Other fiscal arrangements -4.7
Expenses Transfers to governments 70.2
Direct program expenses Transfer payments 45.4
Direct program expenses Capital amortization 5.5
Direct program expenses Operating expenses 88.3
Expenses Direct program expenses 139.1
Expenses Public debt charges 24.7
Revenue Expenses 305.4
Deficit Expenses 5.4
source target value
Personal income tax Income taxes 161.4
Corporate income tax Income taxes 47.3
Non-resident income tax Income taxes 8.3
Income taxes Tax Revenue 217
Goods and Services Tax Excise taxes and duties 37.7
Customs import duties Excise taxes and duties 5.5
Other excise taxes/duties Excise taxes and duties 12.1
Excise taxes and duties Tax Revenue 55.4
Tax Revenue Revenue 272.3
EI premiums Revenue 21.7
Crown corporations Other Revenue 6.6
Other programs Other Revenue 20.6
Net foreign exchange Other Revenue 2.1
Other Revenue Revenue 29.4
Transfers to persons Elderly benefits 53.6
Transfers to persons EI benefits 20.7
Transfers to persons Childrens benefits 23.7
Expenses Transfers to persons 98.1
Transfers to governments Canada Health Transfer 38.6
Transfers to governments Canada Social Transfer 14.2
Transfers to governments Equalization 19
Transfers to governments Territorial Formula Financing 3.8
Transfers to governments Gas Tax Fund2 2.2
Transfers to governments Home care and mental health 0.9
Transfers to governments Other fiscal arrangements -4.9
Expenses Transfers to governments 73.6
Direct program expenses Transfer payments 47.5
Direct program expenses Operating expenses 93.1
Expenses Direct program expenses 140.5
Expenses Public debt charges 26.3
Revenue Expenses 323.4

Sankey diagrams for Federal Budgets for 2016 & 2017

<!DOCTYPE html>
<meta charset="utf-8">
<title>Federal Budget 2017/2018</title>
<style>
.node rect {
cursor: move;
fill-opacity: .9;
shape-rendering: crispEdges;
}
.node text {
pointer-events: none;
text-shadow: 0 1px 0 #fff;
}
.link {
fill: none;
stroke: #000;
stroke-opacity: .2;
}
.link:hover {
stroke-opacity: .5;
}
#sankey-labels {
display: flex;
flex-direction: row;
align-self: center;
max-width: 900px;
justify-content: flex-start;
font-weight: bold;
font-size: 12px;
color: grey;
width: 100%;
margin-top: 8px;
margin-bottom: 8px;
margin-right: 20px;
margin-left: 20px;
}
.sankey-label {
flex: 1;
max-width: 80px;
color: #999;
font-size: 14px;
cursor: pointer;
float: left;
padding: 10px 18px;
border-top: solid 1px #CCC;
border-bottom: solid 1px #CCC;
border-left: solid 1px #CCC;
background: #f9f9f9;
margin: 0 0;
}
.sankey-label.clicked {
color: #000;
background: #e9e9e9;
border-color: #AAA;
box-shadow: inset 0px 0px 4px rgba(0,0,0,0.2);
}
.sankey-label.first {
border-radius: 4px 0 0 4px;
}
.sankey-label.last {
border-right: solid 1px #CCC;
border-radius: 0 4px 4px 0;
}
</style>
<body>
<div id="sankey-labels">
<h4 class="sankey-label first" id="budget2016-button">2016</h4> <!-- &#10132 -->
<h4 class="sankey-label clicked" id="budget2017-button">2017</h4>
<h4 class="sankey-label clicked" id="budget2018-button">2018</h4>
<h4 class="sankey-label last" id="diff-button">Difference</h4>
</div>
<p id="chart">
<script src="http://d3js.org/d3.v3.min.js"></script>
<script src="sankey.js"></script>
<script>
var units = "$B";
var margin = {top: 10, right: 10, bottom: 10, left: 10},
width = 1400 - margin.left - margin.right,
height = 600 - margin.top - margin.bottom;
var formatNumber = d3.format(",.1f"), // zero decimal places
format = function(d) { return formatNumber(d) + " " + units; },
color = d3.scale.category20();
// append the svg canvas to the page
var svg = d3.select("#chart").append("svg")
.attr("width", width + margin.left + margin.right)
.attr("height", height + margin.top + margin.bottom)
.append("g")
.attr("transform",
"translate(" + margin.left + "," + margin.top + ")");
// Set the sankey diagram properties
var sankey = d3.sankey()
.nodeWidth(9)
.nodePadding(10)
.size([width, height]);
var path = sankey.link();
var budget2016, budget2017, budget2018;
// load the data (using the timelyportfolio csv method)
d3.csv("2016.csv", function(error, data) {
budget2016 = data;
d3.csv("2017.csv", function(error, data) {
budget2017 = data;
d3.csv("2018.csv", function(error, data) {
budget2018 = data;
renderBudget(budget2018);
});
});
});
function renderBudget(data){
//set up graph in same style as original example but empty
graph = {"nodes" : [], "links" : []};
data.forEach(function (d) {
graph.nodes.push({ "name": d.source });
graph.nodes.push({ "name": d.target });
graph.links.push({ "source": d.source,
"target": d.target,
"value": +d.value });
});
// return only the distinct / unique nodes
graph.nodes = d3.keys(d3.nest()
.key(function (d) { return d.name; })
.map(graph.nodes));
// loop through each link replacing the text with its index from node
graph.links.forEach(function (d, i) {
graph.links[i].source = graph.nodes.indexOf(graph.links[i].source);
graph.links[i].target = graph.nodes.indexOf(graph.links[i].target);
});
//now loop through each nodes to make nodes an array of objects
// rather than an array of strings
graph.nodes.forEach(function (d, i) {
graph.nodes[i] = { "name": d };
});
sankey
.nodes(graph.nodes)
.links(graph.links)
.layout(32);
// add in the links
svg.selectAll("g").remove();
var link = svg.append("g").selectAll(".link")
.data(graph.links)
.enter().append("path")
.attr("class", "link")
.attr("d", path)
.style("stroke-width", function(d) { return Math.max(1, d.dy); })
.sort(function(a, b) { return b.dy - a.dy; });
// add the link titles
link.append("title")
.text(function(d) {
return d.source.name + " → " +
d.target.name + "\n" + format(d.value); });
// add in the nodes
var node = svg.append("g").selectAll(".node")
.data(graph.nodes)
.enter().append("g")
.attr("class", "node")
.attr("transform", function(d) {
return "translate(" + d.x + "," + d.y + ")"; })
.call(d3.behavior.drag()
.origin(function(d) { return d; })
.on("dragstart", function() {
this.parentNode.appendChild(this); })
.on("drag", dragmove));
// add the rectangles for the nodes
node.append("rect")
.attr("height", function(d) { return d.dy; })
.attr("width", sankey.nodeWidth())
.style("fill", function(d) {
return d.color = color(d.name.replace(/ .*/, "")); })
.style("stroke", function(d) {
return d3.rgb(d.color).darker(2); })
.append("title")
.text(function(d) {
return d.name + "\n" + format(d.value); });
// add in the title for the nodes
node.append("text")
.attr("x", -6)
.attr("y", function(d) { return d.dy / 2; })
.attr("dy", ".35em")
.attr("text-anchor", "end")
.attr("transform", null)
.text(function(d) { return d.name; })
.filter(function(d) { return d.x < width / 2; })
.attr("x", 6 + sankey.nodeWidth())
.attr("text-anchor", "start");
// the function for moving the nodes
function dragmove(d) {
d3.select(this).attr("transform",
"translate(" + d.x + "," + (
d.y = Math.max(0, Math.min(height - d.dy, d3.event.y))
) + ")");
sankey.relayout();
link.attr("d", path);
}
}
d3.select('#budget2016-button').on('click', function () {
d3.selectAll(".sankey-label").classed("clicked", false);
d3.select(this).classed("clicked", true);
renderBudget(budget2016);
});
d3.select('#budget2017-button').on('click', function () {
d3.selectAll(".sankey-label").classed("clicked", false);
d3.select(this).classed("clicked", true);
renderBudget(budget2017);
});
d3.select('#budget2018-button').on('click', function () {
d3.selectAll(".sankey-label").classed("clicked", false);
d3.select(this).classed("clicked", true);
renderBudget(budget2018);
});
d3.select('#diff-button').on('click', function () {
d3.selectAll(".sankey-label").classed("clicked", false);
d3.select(this).classed("clicked", true);
var difference = compare(budget2018, budget2017);
renderBudget(difference);
});
function compare( b2017, b2016){
var diff =[];
for( var i =0 in b2017){
var line = b2017[i];
var diffvalue = 0;
for( var j =0 in b2016){
var line2 = b2016[j];
if( line.source == line2.source && line.target == line2.target){
diffvalue = line.value - line2.value;
}
}
if( diffvalue > 0 ){
var nline = {source:line.source, target:line.target,value:diffvalue};
diff.push(nline);
}
}
return diff;
}
</script>
</body>
</html>
d3.sankey = function() {
var sankey = {},
nodeWidth = 24,
nodePadding = 8,
size = [1, 1],
nodes = [],
links = [];
sankey.nodeWidth = function(_) {
if (!arguments.length) return nodeWidth;
nodeWidth = +_;
return sankey;
};
sankey.nodePadding = function(_) {
if (!arguments.length) return nodePadding;
nodePadding = +_;
return sankey;
};
sankey.nodes = function(_) {
if (!arguments.length) return nodes;
nodes = _;
return sankey;
};
sankey.links = function(_) {
if (!arguments.length) return links;
links = _;
return sankey;
};
sankey.size = function(_) {
if (!arguments.length) return size;
size = _;
return sankey;
};
sankey.layout = function(iterations) {
computeNodeLinks();
computeNodeValues();
computeNodeBreadths();
computeNodeDepths(iterations);
computeLinkDepths();
return sankey;
};
sankey.relayout = function() {
computeLinkDepths();
return sankey;
};
sankey.link = function() {
var curvature = .5;
function link(d) {
var x0 = d.source.x + d.source.dx,
x1 = d.target.x,
xi = d3.interpolateNumber(x0, x1),
x2 = xi(curvature),
x3 = xi(1 - curvature),
y0 = d.source.y + d.sy + d.dy / 2,
y1 = d.target.y + d.ty + d.dy / 2;
return "M" + x0 + "," + y0
+ "C" + x2 + "," + y0
+ " " + x3 + "," + y1
+ " " + x1 + "," + y1;
}
link.curvature = function(_) {
if (!arguments.length) return curvature;
curvature = +_;
return link;
};
return link;
};
// Populate the sourceLinks and targetLinks for each node.
// Also, if the source and target are not objects, assume they are indices.
function computeNodeLinks() {
nodes.forEach(function(node) {
node.sourceLinks = [];
node.targetLinks = [];
});
links.forEach(function(link) {
var source = link.source,
target = link.target;
if (typeof source === "number") source = link.source = nodes[link.source];
if (typeof target === "number") target = link.target = nodes[link.target];
source.sourceLinks.push(link);
target.targetLinks.push(link);
});
}
// Compute the value (size) of each node by summing the associated links.
function computeNodeValues() {
nodes.forEach(function(node) {
node.value = Math.max(
d3.sum(node.sourceLinks, value),
d3.sum(node.targetLinks, value)
);
});
}
// Iteratively assign the breadth (x-position) for each node.
// Nodes are assigned the maximum breadth of incoming neighbors plus one;
// nodes with no incoming links are assigned breadth zero, while
// nodes with no outgoing links are assigned the maximum breadth.
function computeNodeBreadths() {
var remainingNodes = nodes,
nextNodes,
x = 0;
while (remainingNodes.length) {
nextNodes = [];
remainingNodes.forEach(function(node) {
node.x = x;
node.dx = nodeWidth;
node.sourceLinks.forEach(function(link) {
nextNodes.push(link.target);
});
});
remainingNodes = nextNodes;
++x;
}
//
moveSinksRight(x);
scaleNodeBreadths((width - nodeWidth) / (x - 1));
}
function moveSourcesRight() {
nodes.forEach(function(node) {
if (!node.targetLinks.length) {
node.x = d3.min(node.sourceLinks, function(d) { return d.target.x; }) - 1;
}
});
}
function moveSinksRight(x) {
nodes.forEach(function(node) {
if (!node.sourceLinks.length) {
node.x = x - 1;
}
});
}
function scaleNodeBreadths(kx) {
nodes.forEach(function(node) {
node.x *= kx;
});
}
function computeNodeDepths(iterations) {
var nodesByBreadth = d3.nest()
.key(function(d) { return d.x; })
.sortKeys(d3.ascending)
.entries(nodes)
.map(function(d) { return d.values; });
//
initializeNodeDepth();
resolveCollisions();
for (var alpha = 1; iterations > 0; --iterations) {
relaxRightToLeft(alpha *= .99);
resolveCollisions();
relaxLeftToRight(alpha);
resolveCollisions();
}
function initializeNodeDepth() {
var ky = d3.min(nodesByBreadth, function(nodes) {
return (size[1] - (nodes.length - 1) * nodePadding) / d3.sum(nodes, value);
});
nodesByBreadth.forEach(function(nodes) {
nodes.forEach(function(node, i) {
node.y = i;
node.dy = node.value * ky;
});
});
links.forEach(function(link) {
link.dy = link.value * ky;
});
}
function relaxLeftToRight(alpha) {
nodesByBreadth.forEach(function(nodes, breadth) {
nodes.forEach(function(node) {
if (node.targetLinks.length) {
var y = d3.sum(node.targetLinks, weightedSource) / d3.sum(node.targetLinks, value);
node.y += (y - center(node)) * alpha;
}
});
});
function weightedSource(link) {
return center(link.source) * link.value;
}
}
function relaxRightToLeft(alpha) {
nodesByBreadth.slice().reverse().forEach(function(nodes) {
nodes.forEach(function(node) {
if (node.sourceLinks.length) {
var y = d3.sum(node.sourceLinks, weightedTarget) / d3.sum(node.sourceLinks, value);
node.y += (y - center(node)) * alpha;
}
});
});
function weightedTarget(link) {
return center(link.target) * link.value;
}
}
function resolveCollisions() {
nodesByBreadth.forEach(function(nodes) {
var node,
dy,
y0 = 0,
n = nodes.length,
i;
// Push any overlapping nodes down.
nodes.sort(ascendingDepth);
for (i = 0; i < n; ++i) {
node = nodes[i];
dy = y0 - node.y;
if (dy > 0) node.y += dy;
y0 = node.y + node.dy + nodePadding;
}
// If the bottommost node goes outside the bounds, push it back up.
dy = y0 - nodePadding - size[1];
if (dy > 0) {
y0 = node.y -= dy;
// Push any overlapping nodes back up.
for (i = n - 2; i >= 0; --i) {
node = nodes[i];
dy = node.y + node.dy + nodePadding - y0;
if (dy > 0) node.y -= dy;
y0 = node.y;
}
}
});
}
function ascendingDepth(a, b) {
return a.y - b.y;
}
}
function computeLinkDepths() {
nodes.forEach(function(node) {
node.sourceLinks.sort(ascendingTargetDepth);
node.targetLinks.sort(ascendingSourceDepth);
});
nodes.forEach(function(node) {
var sy = 0, ty = 0;
node.sourceLinks.forEach(function(link) {
link.sy = sy;
sy += link.dy;
});
node.targetLinks.forEach(function(link) {
link.ty = ty;
ty += link.dy;
});
});
function ascendingSourceDepth(a, b) {
return a.source.y - b.source.y;
}
function ascendingTargetDepth(a, b) {
return a.target.y - b.target.y;
}
}
function center(node) {
return node.y + node.dy / 2;
}
function value(link) {
return link.value;
}
return sankey;
};
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment