Python 列表：對列表中的值求和並刪除重復項

Question

我在 Python 中有以下列表：

my_list = [{'code_id': 'A', 'amount': 100.0},{'code_id': 'B', 'amount': 150.0},{'code_id': 'C', 'amount': 200.0},{'code_id': 'A', 'amount': 120.0},{'code_id': 'B', 'amount': 300.0},{'code_id': 'D', 'amount': 180.0}]

從上面的列表中，我需要創建一個沒有重復的新列表； 但將上面列表中所有項目的“數量”相加。

我需要實現最終列表，例如：

final_list = [{'code_id': 'A', 'amount': 220.0},{'code_id': 'B', 'amount': 450.0},{'code_id': 'C', 'amount': 200.0},{'code_id': 'D', 'amount': 180.0}]

我能夠刪除重復項，但不知道如何在此過程中對值求和。 我使用的示例代碼：

final_list = []
seen = set()
for dic in my_list:
    key = (dic['code_id'])
    if key in seen:
        continue

    final_list.append(dic)
    seen.add(key)

我怎樣才能在 Python 中實現這一點？

Answer 1

我將從一個臨時defaultdict開始，它跟蹤每個'code_id' 'amount'

from collections import defaultdict  

my_list = [{'code_id': 'A', 'amount': 100.0},{'code_id': 'B', 'amount': 150.0},{'code_id': 'C', 'amount': 200.0}, {'code_id': 'A', 'amount': 120.0},{'code_id': 'B', 'amount': 300.0},{'code_id': 'D', 'amount': 180.0}]          
tmp = defaultdict(int)                                                                                           
                                                                                                                  
for d in my_list: 
    tmp[d['code_id']] += d['amount']
    # if tmp was a normal dict, you could use
    # tmp[d['code_id']] = tmp.get(d['code_id'], 0) + d['amount']
                                                                                                                
print(tmp)                                                                                                              
# defaultdict(int, {'A': 220.0, 'B': 450.0, 'C': 200.0, 'D': 180.0})

...然后轉換tmp的結構以達到所需的結果

result = [{'code_id': k, 'amount': v} for k, v in tmp.items()]                                                   
print(result)                                                                                                    
# [{'code_id': 'A', 'amount': 220.0}, {'code_id': 'B', 'amount': 450.0}, {'code_id': 'C', 'amount': 200.0}, {'code_id': 'D', 'amount': 180.0}]

對於pandas用戶：

>>> pd.DataFrame(my_list).groupby('code_id', as_index=False).sum().to_dict(orient='records')                         
[{'code_id': 'A', 'amount': 220.0},
 {'code_id': 'B', 'amount': 450.0},
 {'code_id': 'C', 'amount': 200.0},
 {'code_id': 'D', 'amount': 180.0}]