帶字典的Defaultdict

Question

我有這樣的字典

a = [{'CohortList': [{'DriverValue': 0.08559936}, {'DriverValue': 0.08184596527051588}], 
      '_id': {'DriverName': 'Yield', 'MonthsOnBooks': 50, 'SegmentName': 'LTV110-Super Prime'}},
     {'CohortList': [{'DriverValue': 2406.04329}, {'DriverValue': 2336.0058100690103}, ], 
      '_id': {'DriverName': 'ADB', 'MonthsOnBooks': 15, 'SegmentName': 'LTV110-Super Prime'}},
     {'CohortList': [{'DriverValue': 2406.04329}, {'DriverValue': 2336.0058100690103}, ], 
      '_id': {'DriverName': 'ADB', 'MonthsOnBooks': 16, 'SegmentName': 'LTV110-Prime'}}]

我想構造一個字典列表，其值是上述字典集合中的列表

    {
    "LTV110-Prime": [
        {
            "ADB": [
                {
                    "16": 1500
                }
            ]
        },
        {
            "Yield": []
        }
    ],
    "LTV110-Super Prime": [
        {
            "ADB": [
                {
                    "15": 1500
                }
            ]
        },
        {
            "Yield": [
                {
                    "50": 0.09
                }
            ]
        }
    ]
}

本質上，我想將每個細分的ADB和Yield與其值分組。

到目前為止，這是我為實現該目標所做的工作。 ADB的值是CohortList列表中DriverValue的平均值。 我使用statistics.mean來找出映射值的平均值。

sg_wrap = defaultdict(dict)
for p in pp_data:
    mapped = map(lambda d: d.get('DriverValue', 0), p['CohortList'])
    dic = {p['_id']['MonthsOnBooks']: statistics.mean(mapped)}

    print(p)
print(sg_wrap)

我無法將驅動程序附加到內部字典。 請幫忙。

Answer 1

由於將所有內容包裝到列表中，因此不需要defaultdict(dcit)而是defaultdict(list) 。

以下似乎有效：

result = defaultdict(list)
for entry in a:
    id_ = entry["_id"]
    name, months, segment = id_["DriverName"], id_["MonthsOnBooks"], id_["SegmentName"]
    values = [x["DriverValue"] for x in entry["CohortList"]]
    d = {name: [{months: statistics.mean(values)}]}
    result[segment].append(d)

結果是

{'LTV110-Prime': [{'ADB': [{16: 2371.0245500345054}]}],
 'LTV110-Super Prime': [{'Yield': [{50: 0.08372266263525793}]},
                        {'ADB': [{15: 2371.0245500345054}]}]}