Python - 将字段和标签添加到嵌套的 json 文件

Question

我有一个数据框如下:

Name_ID | URL                    | Count | Rating
------------------------------------------------
ABC     | www.example.com/ABC    | 10    | 5
123     | www.example.com/123    | 9     | 4
XYZ     | www.example.com/XYZ    | 5     | 2
ABC111  | www.example.com/ABC111 | 5     | 2
ABC121  | www.example.com/ABC121 | 5     | 2
222     | www.example.com/222    | 5     | 3
abc222  | www.example.com/abc222 | 4     | 2
ABCaaa  | www.example.com/ABCaaa | 4     | 2

我正在尝试创建一个 JSON，如下所示:

{
    "name": "sampledata",
    "children": [
        {
            "name": 9,
            "children": [
                {
                    "name": 4,
                    "children": [
                        {
                            "name": "123",
                            "size": 100
                        }
                    ]
                }
            ]
        },
        {
            "name": 10,
            "children": [
                {
                    "name": 5,
                    "children": [
                        {
                            "name": "ABC",
                            "size": 100
                        }
                    ]
                }
            ]
        },
        {
            "name": 4,
            "children": [
                {
                    "name": 2,
                    "children": [
                        {
                            "name": "abc222",
                            "size": 50
                        },
                        {
                            "name": "ABCaaa",
                            "size": 50
                        }
                    ]
                }
            ]
        },
        {
            "name": 5,
            "children": [
                {
                    "name": 2,
                    "children": [
                        {
                            "name": "ABC",
                            "size": 16
                        },
                        {
                            "name": "ABC111",
                            "size": 16
                        },
                        {
                            "name": "ABC121",
                            "size": 16
                        }
                    ]
                },
                {
                    "name": 3,
                    "children": [
                        {
                            "name": "222",
                            "size": 50
                        }
                    ]
                }
            ]
        }
    ]
}

为了做到这一点:

• 我正在尝试在创建 json 时将 "name" 和 "children" 等标签添加到它。

我试过类似的东西

results = [{"name": i, "children": j} for i,j in results.items()]

但我相信它不会正确标记它。

• 此外，添加另一个带有标签“size”的字段，我计划根据公式计算它:

  (Rating*Count*10000)/number_of_children_to_the_immediate_parent
  

这是我的脏代码:

import pandas as pd
from collections import defaultdict
import json

data =[('ABC', 'www.example.com/ABC', 10   , 5), ('123', 'www.example.com/123', 9, 4), ('XYZ', 'www.example.com/XYZ', 5, 2), ('ABC111', 'www.example.com/ABC111', 5, 2), ('ABC121', 'www.example.com/ABC121', 5, 2), ('222', 'www.example.com/222', 5, 3), ('abc222', 'www.example.com/abc222', 4, 2), ('ABCaaa', 'www.example.com/ABCaaa', 4, 2)]

df = pd.DataFrame(data, columns=['Name', 'URL', 'Count', 'Rating'])

gp = df.groupby(['Count'])

dict_json = {"name": "flare"}
children = []

for name, group in gp:
    temp = {}
    temp["name"] = name
    temp["children"] = []

    rgp = group.groupby(['Rating'])
    for n, g in rgp:
        temp2 = {}
        temp2["name"] = n
        temp2["children"] = g.reset_index().T.to_dict().values()
        for t in temp2["children"]:
            t["size"] = (t["Rating"] * t["Count"] * 10000) / len(temp2["children"])
            t["name"] = t["Name"]
            del t["Count"]
            del t["Rating"]
            del t["URL"]
            del t["Name"]
            del t["index"]
        temp["children"].append(temp2)
    children.append(temp)

dict_json["children"] = children

print json.dumps(dict_json, indent=4)

虽然上面的代码确实打印了我需要的东西，但我正在寻找更有效和更简洁的方法来做同样的事情，主要是因为实际的数据集可能更加嵌套和复杂。任何帮助/建议将不胜感激。

Answer 1

非常有趣的问题，也是一个很好的问题!

您可以通过重新组织循环内的代码并使用 list comprehensions 来改进您的方法。 .无需删除内容并在循环内引入临时变量:

dict_json = {"name": "flare"}

children = []
for name, group in gp:
    temp = {"name": name, "children": []}

    rgp = group.groupby(['Rating'])
    for n, g in rgp:
        temp["children"].append({
            "name": n,
            "children": [
                {"name": row["Name"],
                 "size": row["Rating"] * row["Count"] * 10000 / len(g)}
                for _, row in g.iterrows()
            ]
        })

    children.append(temp)

dict_json["children"] = children

或者，“包装”版本:

dict_json = {
    "name": "flare", 
    "children": [
        {
            "name": name, 
            "children": [
                {
                    "name": n,
                    "children": [
                        {
                            "name": row["Name"],
                            "size": row["Rating"] * row["Count"] * 10000 / len(g)
                        } for _, row in g.iterrows()
                    ]
                } for n, g in group.groupby(['Rating'])
            ]
        } for name, group in gp
    ]
}

我正在为您打印以下字典示例输入数据框:

{
    "name": "flare", 
    "children": [
        {
            "name": 4, 
            "children": [
                {
                    "name": 2, 
                    "children": [
                        {
                            "name": "abc222", 
                            "size": 40000
                        }, 
                        {
                            "name": "ABCaaa", 
                            "size": 40000
                        }
                    ]
                }
            ]
        }, 
        {
            "name": 5, 
            "children": [
                {
                    "name": 2, 
                    "children": [
                        {
                            "name": "XYZ", 
                            "size": 33333
                        }, 
                        {
                            "name": "ABC111", 
                            "size": 33333
                        }, 
                        {
                            "name": "ABC121", 
                            "size": 33333
                        }
                    ]
                }, 
                {
                    "name": 3, 
                    "children": [
                        {
                            "name": "222", 
                            "size": 150000
                        }
                    ]
                }
            ]
        }, 
        {
            "name": 9, 
            "children": [
                {
                    "name": 4, 
                    "children": [
                        {
                            "name": "123", 
                            "size": 360000
                        }
                    ]
                }
            ]
        }, 
        {
            "name": 10, 
            "children": [
                {
                    "name": 5, 
                    "children": [
                        {
                            "name": "ABC", 
                            "size": 500000
                        }
                    ]
                }
            ]
        }
    ]
}