遍历对象数组以生成d3 Sankey图表数据

// 假设有一个包含对象的数组，每个对象包含节点和连接信息
const data = [
  { source: "A", target: "B", value: 10 },
  { source: "B", target: "C", value: 5 },
  { source: "A", target: "C", value: 8 },
  // 更多对象...
];

// 创建一个空数组，用于存储生成的d3 Sankey图表数据
const sankeyData = [];

// 遍历对象数组
data.forEach((obj) => {
  // 提取节点和连接信息
  const { source, target, value } = obj;

  // 检查节点是否已存在于sankeyData中
  let sourceNode = sankeyData.find((node) => node.name === source);
  let targetNode = sankeyData.find((node) => node.name === target);

  // 如果节点不存在，则创建新节点并添加到sankeyData中
  if (!sourceNode) {
    sourceNode = { name: source };
    sankeyData.push(sourceNode);
  }
  if (!targetNode) {
    targetNode = { name: target };
    sankeyData.push(targetNode);
  }

  // 创建连接信息并添加到节点的连接数组中
  const link = { source: sourceNode, target: targetNode, value };
  sourceNode.targets = sourceNode.targets || [];
  sourceNode.targets.push(link);
  targetNode.sources = targetNode.sources || [];
  targetNode.sources.push(link);
});

// 最终生成的sankeyData数据格式如下：
// [
//   { name: "A", targets: [{ source: {...}, target: {...}, value: 10 }, ...] },
//   { name: "B", sources: [{ source: {...}, target: {...}, value: 10 }, ...], targets: [...] },
//   { name: "C", sources: [...], targets: [...] },
//   ...
// ]