Как преобразовать иерархическую таблицу в JSON - PullRequest
Новая
       12

Как преобразовать иерархическую таблицу в JSON

0 голосов
/ 20 декабря 2018

Я пытаюсь преобразовать многоуровневую таблицу иерархии в определенный формат JSON для создаваемого мной визуала.

У меня есть данные в кадре данных pandas, и я попытался сгруппировать их по различным уровням,но затем не может преобразовать группу в JSON с помощью панд.Я также попытался просто преобразовать фрейм данных в формат 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 to_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" 
        },
      ],
    },
    ],
  },
]})

1 Ответ

0 голосов
/ 20 декабря 2018

Вот и мы 

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))
...