D3:使用嵌套函数将父键的平面数据转换为层次结构

时间:2012-08-13 13:07:22

标签: javascript json nested operators d3.js

我确信有一种非常简单的优雅方式可以做到这一点,但我无法弄明白。我有一些看起来像这样的输入数据:

[
{id: 1, name: "Peter"},
{id: 2, name: "Paul", manager: 1},
{id: 3, name: "Mary", manager: 1},
{id: 4, name: "John", manager: 2},
{id: 5, name: "Jane", manager: 2}
]

如果可能,我想使用d3.js嵌套操作符来获取在层次结构布局中使用的结构。像这样:

[ 
   {name: "Peter", children: [
          {name:"Paul", children: [
              {name:"John"},
              {name:"Jane"}
          ]},
          {name:"Mary"}
      ]
   }
]

1 个答案:

答案 0 :(得分:12)

您不能在此处使用嵌套运算符,因为嵌套会生成固定的层次结构:输出层次结构中的级别数与您指定的键函数数相同。

也就是说,您可以编写自己的函数来生成树。假设根节点是输入数组中的第一个节点,您可以从id到node创建一个映射,然后懒惰地构造树。

function tree(nodes) {
  var nodeById = {};

  // Index the nodes by id, in case they come out of order.
  nodes.forEach(function(d) {
    nodeById[d.id] = d;
  });

  // Lazily compute children.
  nodes.forEach(function(d) {
    if ("manager" in d) {
      var manager = nodeById[d.manager];
      if (manager.children) manager.children.push(d);
      else manager.children = [d];
    }
  });

  return nodes[0];
}

如果您知道节点按顺序列出,以便管理员在报告之前显示,则可以简化代码以仅迭代一次。