Public
Edited
Sep 15, 2023
Importers
Insert cell
Insert cell
data = FileAttachment("sp500-components-2023-09-10.json").json()
Insert cell
treemapOptions = Object.keys(d3).filter(d => d.startsWith('treemap') && d !== "treemap");
Insert cell
viewof treemapSelected = Inputs.select(treemapOptions, {value: treemapOptions[5], label: "Treemap types"})
Insert cell
treemapType = treemapSelected
Insert cell
import {interpolateCategoricalOptions as schemeOptions} from "@chiahsun-ws/d3-functions"
Insert cell
viewof schemeSelected = Inputs.select(schemeOptions, {value: schemeOptions[9], label: "Scheme colors"})
Insert cell
schemeColor = schemeSelected
Insert cell
nameColumn = "name"
Insert cell
valueColumn = "value"
Insert cell
Insert cell
chart = {
const margin = {top: 0, right: 0, bottom: 0, left: 0};

const svg_width = 2000;
const svg_height = 1000;

const width = svg_width - margin.left - margin.right;
const height = svg_height - margin.top - margin.bottom;

const svg = d3.create("svg")
.attr("width", svg_width)
.attr("height", svg_height)
.attr("viewBox", [0, 0, width + margin.left + margin.right, height + margin.top + margin.bottom])
.attr("style", "max-width: 100%; height: auto; font: 20px sans-serif;");

const chart = svg.append("g")
.attr("transform", `translate(${margin.left}, ${margin.top})`);

const tile = d3[treemapType];

const color = d3.scaleOrdinal(data.children.map(d => d[nameColumn]), d3[schemeColor]);
const root = d3.treemap()
.tile(tile)
.size([width, height])
.padding(1)
(d3.hierarchy(data)
.sum(d => d[valueColumn])
.sort((a, b) => b[valueColumn] - a[valueColumn]));

const leaf = chart.selectAll("g")
.data(root.leaves())
.join("g")
.attr("transform", d => `translate(${d.x0}, ${d.y0})`);

leaf.append("rect")
.attr("id", d => (d.leafUid = DOM.uid("leaf")).id)
.attr("fill", d => { while (d.depth > 1) d = d.parent; return color(d.data[nameColumn]); })
.attr("fill-opacity", 0.6)
.attr("width", d => d.x1 - d.x0)
.attr("height", d => d.y1 - d.y0);

const format = d3.format(",d");

leaf.append("clipPath")
.attr("id", d => (d.clipUid = DOM.uid("clip")).id)
.append("use")
.attr("xlink:href", d => d.leafUid.href);
leaf.append("text")
.attr("clip-path", d => d.clipUid)
.selectAll("tspan")
.data(d => d.data[nameColumn].split(/(?=[A-Z][a-z])|\s+/g).concat(format(d[valueColumn])))
.join("tspan")
.attr("x", 3)
.attr("y", (d, i, nodes) => `${(i === nodes.length-1) * 0.3 + 1.1 + 0.9 * i}em`)
.attr("fill-opacity", (d, i, nodes) => (i === nodes.length-1) ? 0.7 : null)
.text(d => d);

leaf.append("title")
.text(d => `${d.ancestors().reverse().map(d => d.data[nameColumn]).join(".")}\n${format(d[valueColumn])}`)
return Object.assign(svg.node(), {scales: {color}});
}
Insert cell

Purpose-built for displays of data

Observable is your go-to platform for exploring data and creating expressive data visualizations. Use reactive JavaScript notebooks for prototyping and a collaborative canvas for visual data exploration and dashboard creation.
Learn more