用d3绘制热图

时间:2012-10-17 00:06:41

标签: javascript d3.js heatmap

我正在尝试使用来自csv的数据绘制带有d3的热图:这是我到目前为止所拥有的

给出一个csv文件:

row,col,score
0,0,0.5
0,1,0.7
1,0,0.2
1,1,0.4

我有一个svg和代码如下:

<svg id="heatmap-canvas" style="height:200px"></svg>

<script>
d3.csv("sgadata.csv", function(data) {
    data.forEach(function(d) { 
    d.score = +d.score;
    d.row = +d.row;
d.col = +d.col;
});
//height of each row in the heatmap
//width of each column in the heatmap
var h = gridSize;
var w = gridSize;
var rectPadding = 60;

$('#heatmap-canvas').empty();

var mySVG = d3.select("#heatmap-canvas")
.style('top',0)
.style('left',0);

var colorScale = d3.scale.linear()
.domain([-1, 0, 1])
.range([colorLow, colorMed, colorHigh]);

rowNest = d3.nest()
.key(function(d) { return d.row; })
.key(function(d) { return d.col; });

dataByRows = rowNest.entries(data);
mySVG.forEach(function(){
var heatmapRow = mySVG.selectAll(".heatmap")
    .data(dataByRows, function(d) { return d.key; })
    .enter().append("g");

    //For each row, generate rects based on columns - this is where I get stuck
    heatmapRow.forEach(function(){
    var heatmapRects = heatmapRow
        .selectAll(".rect")
        .data(function(d) {return d.score;})
        .enter().append("svg:rect")
                    .attr('width',w)
        .attr('height',h)
        .attr('x', function(d) {return (d.row * w) + rectPadding;})
        .attr('y', function(d) {return (d.col * h) + rectPadding;})
        .style('fill',function(d) {
            if(d.score == NaN){return colorNA;}
            return colorScale(d.score);
                 })

})
</script>

我的问题在于嵌套。我的嵌套基于2个键,第一行(用于生成行),然后对于每一行,列有多个嵌套键,每个键都包含我的分数。 我不知道如何继续,即循环遍历列并添加颜色

的矩形

任何帮助将不胜感激

1 个答案:

答案 0 :(得分:10)

虽然您可以使用subselect(请参阅d3.js building a grid of rectangles)来处理d3中的嵌套数据,但在这种情况下并不需要它。我在http://jsfiddle.net/QWLkR/2/使用您的数据汇集了一个示例。这是关键部分:

var heatMap = svg.selectAll(".heatmap")
    .data(data, function(d) { return d.col + ':' + d.row; })
  .enter().append("svg:rect")
    .attr("x", function(d) { return d.row * w; })
    .attr("y", function(d) { return d.col * h; })
    .attr("width", function(d) { return w; })
    .attr("height", function(d) { return h; })
    .style("fill", function(d) { return colorScale(d.score); });

基本上,您只需使用rowcol来计算热图中方块的正确位置。