🎉 I'm releasing 12 products in 12 months! If you love product, checkout my new blog workingoutloud.dev

Back to home

Build Your Own Heat Map with D3

Heat maps are a great way to display correlations between two sets of data or quickly communicating progress on a project (think node.green).

I recently had to draw up some heat maps at work to track progress on some projects I am working on with the power of D3. We are going to implement a short look at generating one with D3.

Note: we are not going to use any frameworks today.

Prerequisites

We are going to use Vercel's serve package to serve our static files.

Follow the link to read more about it.

# Install serve globally npm i -g serve

Getting Started

mkdir d3-heatmap cd d3-heatmap touch index.html main.css main.js

The styles file

Let's add some CSS to our main.css file.

text { font-size: 10px; font-family: "Roboto Mono", monospace; font-weight: 700; } line, path { fill: none; stroke: #000; shape-rendering: crispEdges; }

As this is a trivial example, we are going to target the HTML. Normally, applying a class is a better idea.

This sets the font to be Roboto Mono (which we will bring in from the Google Fonts CDN) and sets some CSS property values for the line and path SVG elements.

The JavaScript

The JavaScript is the main place where the magic happens.

Let's add the following to main.js. I will add comments in the code about what is happening.

// Assign a 2d array of correlating values. // This each subarray will render as a row const data = [ [1, 1, 1, 1], [1, 0.8, 1, 0.5], [0, 1, 1, 1], [1, 1, 1, 0], ]; // Add our labels as an array of strings const rowLabelsData = ["First Row", "Second Row", "Third Row", "Fourth Row"]; const columnLabelsData = [ "First Column", "Second Column", "Third Column", "Fourth Column", ]; function Matrix(options) { // Set some base properties. // Some come from an options object // pass when `Matrix` is called. const margin = { top: 50, right: 50, bottom: 180, left: 180 }, width = 350, height = 350, container = options.container, startColor = options.start_color, endColor = options.end_color; // Find our max and min values const maxValue = d3.max(data, (layer) => { return d3.max(layer, (d) => { return d; }); }); const minValue = d3.min(data, (layer) => { return d3.min(layer, (d) => { return d; }); }); const numrows = data.length; // assume all subarrays have same length const numcols = data[0].length; // Create the SVG container const svg = d3 .select(container) .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 + ")"); // Add a background to the SVG const background = svg .append("rect") .style("stroke", "black") .attr("width", width) .attr("height", height); // Build some scales for us to use const x = d3.scale.ordinal().domain(d3.range(numcols)).rangeBands([0, width]); const y = d3.scale .ordinal() .domain(d3.range(numrows)) .rangeBands([0, height]); // This scale in particular will // scale our colors from the start // color to the end color. const colorMap = d3.scale .linear() .domain([minValue, maxValue]) .range([startColor, endColor]); // Generate rows and columns and add // color fills. const row = svg .selectAll(".row") .data(data) .enter() .append("g") .attr("class", "row") .attr("transform", (d, i) => { return "translate(0," + y(i) + ")"; }); const cell = row .selectAll(".cell") .data((d) => { return d; }) .enter() .append("g") .attr("class", "cell") .attr("transform", (d, i) => { return "translate(" + x(i) + ", 0)"; }); cell .append("rect") .attr("width", x.rangeBand() - 0.3) .attr("height", y.rangeBand() - 0.3); row .selectAll(".cell") .data((d, i) => { return data[i]; }) .style("fill", colorMap); const labels = svg.append("g").attr("class", "labels"); const columnLabels = labels .selectAll(".column-label") .data(columnLabelsData) .enter() .append("g") .attr("class", "column-label") .attr("transform", (d, i) => { return "translate(" + x(i) + "," + height + ")"; }); columnLabels .append("line") .style("stroke", "black") .style("stroke-width", "1px") .attr("x1", x.rangeBand() / 2) .attr("x2", x.rangeBand() / 2) .attr("y1", 0) .attr("y2", 5); columnLabels .append("text") .attr("x", 0) .attr("y", y.rangeBand() / 2 + 20) .attr("dy", ".82em") .attr("text-anchor", "end") .attr("transform", "rotate(-60)") .text((d, i) => { return d; }); const rowLabels = labels .selectAll(".row-label") .data(rowLabelsData) .enter() .append("g") .attr("class", "row-label") .attr("transform", (d, i) => { return "translate(" + 0 + "," + y(i) + ")"; }); rowLabels .append("line") .style("stroke", "black") .style("stroke-width", "1px") .attr("x1", 0) .attr("x2", -5) .attr("y1", y.rangeBand() / 2) .attr("y2", y.rangeBand() / 2); rowLabels .append("text") .attr("x", -8) .attr("y", y.rangeBand() / 2) .attr("dy", ".32em") .attr("text-anchor", "end") .text((d, i) => { return d; }); }

The HTML file

Inside of index.html, add the following.

<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8" /> <meta name="viewport" content="width=device-width, initial-scale=1.0" /> <title>Heatmap Example</title> <link rel="stylesheet" type="text/css" href="main.css" /> <link href="https://fonts.googleapis.com/css2?family=Roboto+Mono:wght@700&display=swap" rel="stylesheet" /> <script type="text/javascript" src="https://cdnjs.cloudflare.com/ajax/libs/d3/3.5.17/d3.min.js" ></script> </head> <body> <div style="display:inline-block; float:left" id="container"></div> <script src="main.js"></script> <script> Matrix({ container: "#container", start_color: "#FC7C89", end_color: "#21A38B", }); </script> </body> </html>

In this file, we are bringing in D3 + a Roboto Mono theme in from CDNs, plus loading out main.css and main.js files.

Finally, we call Matrix with the options object that we wrote in the JS file.

Running

Within our work directory, run serve . - this will serve the files on port 5000.

Alternatively, you could just run open index.html to open the file in the default browser. I only added serve to this tutorial as I use it all the time to serve more complex builds.

If we open up http://localhost:5000 we will see our heat map.

Final heatmap

Final heatmap

Resources and Further reading

  1. Google Fonts
  2. Serve
  3. Node Green
  4. D3 Heatmap

Image credit: Anqi Lu

Personal image

Dennis O'Keeffe

@dennisokeeffe92
  • Melbourne, Australia

Hi, I am a professional Software Engineer. Formerly of Culture Amp, UsabilityHub, Present Company and NightGuru.
I am currently working on Visibuild.

1,200+ PEOPLE ALREADY JOINED ❤️️

Get fresh posts + news direct to your inbox.

No spam. We only send you relevant content.

Build Your Own Heat Map with D3

Introduction

Share this post