Python3：获取字典列表中特定键值的总和[重复]

Question

我遇到了一个我无法弄清楚的简单问题。
我有一个保存在 json 文件中的对象列表。一旦转换成 Python，它们就会给我这样的东西：

data = [
    {"id": 1, "value": 3},
    {"id": 2, "value": 1},
    {"id": 3, "value": 5},
    {"id": 1, "value": 1},
    {"id": 1, "value": 2},
    {"id": 3, "value": 2},
    {"id": 1, "value": 3}
]

我试图每个唯一的“id”只有一个对象/字典。换句话说，我试图获得以下结果：

[{"id": 1, "value": 9}, {"id": 2, "value": 1}, {"id": 3, "value": 7}]

显然，我可以做到这一点：

foo = list(set([item["id"] for item in data]))

newList = []
for i in foo:
    bar = {"id": i, "value": 0}
    for x in data:
        if x['id'] == i:
            bar['value'] += x['value']

    newList.append(bar)

但是我担心这个三重嵌套循环可能会大大减慢处理大型数据集的过程。

我对@9​​87654325@ 很熟悉，我在collection's Counters 上看到过这个问题，但这些对我来说不起作用。

任何想法如何获得所需的结果？

Answer 1

Pandas 来救援！

首先，根据您的数据创建一个数据框

df = pd.DataFrame(data)
print(df)

Output:
   id  value
0   1      3
1   2      1
2   3      5
3   1      1
4   1      2
5   3      2
6   1      3

然后，group by id 和 sum

result = df.groupby("id").sum()
print(result)

Output:
    value
id       
1       9
2       1
3       7

然后将其转换为您想要的列表：

new_list = [{"id": elemid, "value": val.value} for elemid, val in result.iterrows()]
print(new_list)
# Output: [{'id': 1, 'value': 9}, {'id': 2, 'value': 1}, {'id': 3, 'value': 7}]

或者，您可以改进现有方式：

data = [
    {"id": 1, "value": 3},
    {"id": 2, "value": 1},
    {"id": 3, "value": 5},
    {"id": 1, "value": 1},
    {"id": 1, "value": 2},
    {"id": 3, "value": 2},
    {"id": 1, "value": 3}
]

new_data = {}
for elem in data:
    elemid = elem["id"]
    value = elem["value"]
    new_data[elemid] = new_data.get(elemid, 0) + value

print(new_data)
# Output: {1: 9, 2: 1, 3: 7}

然后，将其转换为您想要的列表：

new_list = [{"id": key, "value": value} for key, value in new_data.items()]
print(new_list)
# Output: [{'id': 1, 'value': 9}, {'id': 2, 'value': 1}, {'id': 3, 'value': 7}]