将平面表转换为层次结构

时间:2021-02-18 08:24:30

标签: javascript d3.js multidimensional-array

我想将表格从 csv 转换为层次结构。表格格式如下:

lvl1 lvl2 lvl3 size
   A    a    1    1
   A    a    2    2
   A    b    1    3
   A    b    2    4
   B    a    1    5
   B    a    2    6
   B    b    1    7
   B    b    2    8

转换为下面的 JSON 对象。

我这样使用 d3.nest

var n = d3.nest()
  .key(d => d.lvl1)
  .key(d => d.lvl2)
  .key(d => d.lvl3)
  .entries(d)

产生以下格式,这几乎是我想传递给 d3.hierarchy 的格式:

[{
    "key": "A",
    "values": [{
      "key": "a",
      "values": [{
          "key": "1",
          "values": [{
            "lvl1": "A",
            "lvl2": "a",
            "lvl3": 1,
            "size": 1
          }]
        },
        {
          "key": "2",
          "values": [{
            "lvl1": "A",
            "lvl2": "a",
            "lvl3": 2,
            "size": 2
          }]
        }
      ]
    }]
  }
  //...
]

但是,这增加了一个级别太多,理想情况下我希望有以下输出:

[{
    "key": "A",
    "values": [{
      "key": "a",
      "values": [{
          "key": "1",
          "size": 1
        },
        {
          "key": "2",
          "size": 2
        }
      ]
    }]
  } //,...
]

我可以方便地将这种格式传递给 d3.hierarchy(n, d => d.values)。如何实现?

注意我不在乎叶子中是否有其他字段,因此,只要叶子不再嵌套,这样的输出也可以:

[{
    "key": "A",
    "values": [{
      "key": "a",
      "values": [{
          "key": "1",
          "lvl1": "A",
          "lvl2": "a",
          "lvl3": 1,
          "size": 1
        },
        {
          "key": "2",
          "lvl1": "A",
          "lvl2": "a",
          "lvl3": 2,
          "size": 2

        }
      ]
    }]
  } //,...
]

JSON

[{
    "lvl1": "A",
    "lvl2": "a",
    "lvl3": 1,
    "size": 1
  },
  {
    "lvl1": "A",
    "lvl2": "a",
    "lvl3": 2,
    "size": 2
  },
  {
    "lvl1": "A",
    "lvl2": "b",
    "lvl3": 1,
    "size": 3
  },
  {
    "lvl1": "A",
    "lvl2": "b",
    "lvl3": 2,
    "size": 4
  },
  {
    "lvl1": "B",
    "lvl2": "a",
    "lvl3": 1,
    "size": 5
  },
  {
    "lvl1": "B",
    "lvl2": "a",
    "lvl3": 2,
    "size": 6
  },
  {
    "lvl1": "B",
    "lvl2": "b",
    "lvl3": 1,
    "size": 7
  },
  {
    "lvl1": "B",
    "lvl2": "b",
    "lvl3": 2,
    "size": 8
  }
]

1 个答案:

答案 0 :(得分:2)

您可以通过 rollup 获得您想要的:

const n = d3.nest()
  .key(d => d.lvl1)
  .key(d => d.lvl2)
  .rollup(d => d.map(e => ({
    key: e.lvl3,
    size: e.size
  })))
  .entries(data);

这是演示:

const data = [{
    "lvl1": "A",
    "lvl2": "a",
    "lvl3": 1,
    "size": 1
  },
  {
    "lvl1": "A",
    "lvl2": "a",
    "lvl3": 2,
    "size": 2
  },
  {
    "lvl1": "A",
    "lvl2": "b",
    "lvl3": 1,
    "size": 3
  },
  {
    "lvl1": "A",
    "lvl2": "b",
    "lvl3": 2,
    "size": 4
  },
  {
    "lvl1": "B",
    "lvl2": "a",
    "lvl3": 1,
    "size": 5
  },
  {
    "lvl1": "B",
    "lvl2": "a",
    "lvl3": 2,
    "size": 6
  },
  {
    "lvl1": "B",
    "lvl2": "b",
    "lvl3": 1,
    "size": 7
  },
  {
    "lvl1": "B",
    "lvl2": "b",
    "lvl3": 2,
    "size": 8
  }
];

var nested = d3.nest()
  .key(d => d.lvl1)
  .key(d => d.lvl2)
  .rollup(d => d.map(e => ({
    key: e.lvl3,
    size: e.size
  })))
  .entries(data);

console.log(nested);
<script src="https://cdnjs.cloudflare.com/ajax/libs/d3/5.7.0/d3.min.js"></script>