如何将分层表格转换为JSON

Question

如何将分层表格转换为JSON

3

我正在尝试将一个多级层次的表格转换成特定的JSON格式，以便创建一个可视化。

我已经将数据存储在一个pandas dataframe中，并尝试按不同的级别进行分组，但是无法使用pandas将groupby转换为json。我也尝试过将dataframe直接转换为json，但格式不正确。我不知道还有什么其他方法可以获得我想要的父/子格式。所有“size”值只需要为1，因此这部分似乎很简单...... 提前感谢您的帮助！

**This is what my data looks like**
ColA     ColB     ColC   
Parent1  Child1   
Parent1  Child2   Child2A 
Parent1  Child2   Child2B
Parent1  Child3   Child2A
Parent2  Child1
Parent2  Child2   Child2A

我从pandas dataframe转换成json时得到的是按列创建json，因此失去了层次结构方面的信息。

所以我得到的是：

data = {"Parent1}"{"index #":"col2 value"

我想要的是：

data = ({ "name":"TEST",
"children": [
  {
    "name": "Parent1",
    "children": 
      [
      {
        "name": "Child1",
        "size": "1"
      },
      {
      "name":"Child2",
        "children": 
        [
        {
          "name":"Child2A",
          "size":"1" 
        },
        {
          "name":"Child2B",
          "size":"1" 
        },
        {
          "name":"Child2C",
          "size":"1" 
        },
        {
          "name":"Child2D",
          "size":"1" 
        },
        ],
      },
    {
      "name":"Parent2",
      "children": [
        {
          "name":"Child2A",
          "size":"1" 
        },
        {
          "name":"Child2B",
          "size":"1" 
        },
        {
          "name":"Child2C",
          "size":"1" 
        },
      ]
    },
    ]
  },
  {
    "name": "Parent3",
    "children": 
    [
      {
        "name": "Child1",
        "size": "1",
      },
      {
      "name":"Child2",
      "children": 
      [
        {
          "name":"Child2A",
          "size":"1" 
        },
        {
          "name":"Child2B",
          "size":"1" 
        },
        {
          "name":"Child2C",
          "size":"1" 
        },
      ],
    },
    {
      "name":"Child3",
      "children": 
      [
        {
          "name":"Child3A",
          "size":"1" 
        },
      ],
    },
    ],
  },
]})

- Katy

没有直接的库/解决方案，首先需要将您的数据框转换为JSON，然后按照您的方式自定义JSON。 - Soumendra

谢谢您的回复，您能提供一些指导，告诉我如何自定义它以使其符合那种格式吗？ - Katy

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- grapes · Accepted Answer

我们来了

import json

data = [
    'Parent1  Child1',
    'Parent1  Child2   Child2A',
    'Parent1  Child2   Child2B',
    'Parent1  Child3   Child2A',
    'Parent2  Child1',
    'Parent2  Child2   Child2A',
]

tree = {}

for d in data:
    node = None
    for item in d.split():
        name = item.strip()  # dont need spaces
        current_dict = tree if node is None else node
        node = current_dict.get(name)
        if not node:
            node = {}
            current_dict[name] = node


def walker(src, res):
    for name, value in src.items():
        node = {'name': name, 'size': 1}
        if 'children' not in res:
            res['children'] = []
        res['children'].append(node)
        walker(value, node)

result = {'name': 'TEST'}
walker(tree, result)

print (json.dumps(result, indent = True))