python - 将属性等于的数组中的对象子数组分组 - Python

Question

我有一个数组，如果键值对等于userID，我将尝试将对象的子数组分组在一起。

给我留下一个对象，每个 userID 以及该 userID 的所有子数组。

即使在搜索了SO之后，我似乎也不知道如何做到这一点。

如何对 userID 相同的子数组进行分组？(数据发生变化，所以我需要使用for循环)

感谢您的帮助。

数组看起来像这样:

      [  
       {  
          'name':'James',
          'lastname':'Bond',
          'userID': 1001,
          'subarray':[  
             {  
                'color':'blue',
                'animal':'dog'
             }
          ]
       },
       {  
          'name':'James',
          'lastname':'Bond',
          'userID': 1001,
          'subarray':[  
             {  
                'color':'red',
                'animal':'cat'
             }
          ]
       },
       {  
          'name':'Billy',
          'lastname':'King',
          'userID': 1004,
          'subarray':[  
             {  
                'color':'green',
                'animal':'fish'
             }
          ]
       }
    ]

我想让数组像这样:

      [  
       {  
          'name':'James',
          'lastname':'Bond',
          'userID': 1001,
          'subarray':[  
             {  
                'color':'blue',
                'animal':'dog'
             },
             {  
                'color':'red',
                'animal':'cat'
             }
          ]
       },
       {  
          'name':'Billy',
          'lastname':'King',
          'userID': 1004,
          'subarray':[  
             {  
                'color':'green',
                'animal':'fish'
             }
          ]
       }
    ]

Answer 1

使用简单的迭代。

例如:

result = {}
for item in data:
    if item["userID"] not in result:
        result[item["userID"]] = {'name':item["name"], 'lastname':item["lastname"],'userID': item["userID"],'subarray':[]}
    result[item["userID"]]['subarray'].append(item["subarray"])

print(list(result.values()))

输出:

[{'lastname': 'Bond',
  'name': 'James',
  'subarray': [[{'animal': 'dog', 'color': 'blue'}],
               [{'animal': 'cat', 'color': 'red'}]],
  'userID': 1001},
 {'lastname': 'King',
  'name': 'Billy',
  'subarray': [[{'animal': 'fish', 'color': 'green'}]],
  'userID': 1004}]