0

Origin data如下图，每个项目都有一个类型标记，比如interests, family, behaviors, etc我想按这个类型字段分组。

return_data = [
{
      "id": "112",
      "name": "name_112",
      "type": "interests",
    },
    {
      "id": "113",
      "name": "name_113",
      "type": "interests",
    },
      {
      "id": "114",
      "name": "name_114",
      "type": "interests",
    },
      {
      "id": "115",
      "name": "name_115",
      "type": "behaviors",
    },
         {
      "id": "116",
      "name": "name_116",
      "type": "family",
    },
         {
      "id": "117",
      "name": "name_117",
      "type": "interests",
    },
    ...
]

expected ouput数据格式如下：

output_data = [

    {"interests":[
        {
          "id": "112",
          "name": "name_112"
        },
        {
          "id": "113",
          "name": "name_113"
        },
        ...
        ]
    },
    {
        "behaviors": [
            {
                "id": "115",
                "name": "name_115"
            },
            ...
        ]
    },
    {
        "family": [
            {
                "id": "116",
                "name": "name_116"
            },
            ...
        ]
    },
    ...
]

这是我的试验：

type_list = []
for item in return_data:
    if item['type'] not in type_list:
        type_list.append(item['type'])

interests_list = []
for type in type_list:
    temp_list = []
    for item in return_data:
        if item['type'] == type:
            temp_list.append({"id": item['id'], "name": item['name']})
    interests_list.append({type: temp_list})

显然我的试验效率很低，因为它是 O(n*m)，但我找不到更有效的方法来解决这个问题。

有没有更有效的方法来获得结果？非常欢迎任何评论，谢谢。

4

2 回答 2

2

使用 adefaultdict存储每种类型的项目列表：

from collections import defaultdict

# group by type
temp_dict = defaultdict(list)
for item in return_data:
    temp_dict[item["type"]].append({"id": item["id"], "name": item["name"]})

# convert back into a list with the desired format
output_data = [{k: v} for k, v in temp_dict.items()]

输出：

[
    {
        'behaviors': [
            {'name': 'name_115', 'id': '115'}
        ]
    }, 
    {
        'family': [
            {'name': 'name_116', 'id': '116'}
        ]
    }, 
    {
        'interests': [
            {'name': 'name_112', 'id': '112'},
            {'name': 'name_113', 'id': '113'},
            {'name': 'name_114', 'id': '114'},
            {'name': 'name_117', 'id': '117'}
        ]
    },
    ...
]

如果您不想导入defaultdict，可以使用带有以下内容的香草字典setdefault：

# temp_dict = {}

temp_dict.setdefault(item["type"], []).append(...)

行为方式完全相同，但效率稍低。

于 2019-03-29T11:09:48.023 回答
0

请参阅Python 字典以获取地图。

for item in return_data:
typeMap[item['type']] = typeMap[item['type']]  + delimiter + item['name']
于 2019-03-29T11:13:26.287 回答