Pythonで辞書のキーごとの平均値(累積平均)を求める方法

概要

複数の辞書に含まれる同じキーの値に対して、累積平均(平均値)を求めたいケースはよくあります。このような場合、まず結果を格納するための空の辞書を作成し、元の辞書のリストを順番に走査しながら各項目(キーと値)へアクセスします。キーがすでに結果用の辞書に存在する場合は、その値をリストに追加していき、存在しない場合は新しいキーとして値をリスト形式で格納します。最後に、statisticsモジュールのmean()関数を使って、各キーごとの平均値を計算します。

サンプルコード

以下に具体的な実装例を示します。

from statistics import mean

my_list = [{'hi' : 24, 'there' : 81, 'how' : 11},
   {'hi' : 16, 'how' : 78, 'doing' : 63}]

print("The list is : ")
print(my_list)

my_result = dict()
for sub in my_list:
   for key, val in sub.items():
      if key in my_result:

         my_result[key].append(val)
      else:
         my_result[key] = [val]

for key, my_val in my_result.items():
   my_result[key] = mean(my_val)

print("The result is : ")
print(my_result)

出力結果

The list is :
[{'hi': 24, 'there': 81, 'how': 11}, {'hi': 16, 'how': 78, 'doing': 63}]
The result is :
{'hi': 20, 'there': 81, 'how': 44.5, 'doing': 63}

コードの解説

• 必要なパッケージ(statisticsモジュール)をインポートします。

• 辞書を要素とするリストを定義し、コンソールに表示します。

• 結果を格納するための空の辞書を定義します。

• 元の辞書リストを反復処理し、各項目(キーと値)を取得します。

• キーがすでに結果辞書に存在する場合は、その値を対応するリストに追加します。

• 存在しない場合は、新しいキーとして値をリスト形式で格納します。

• 再度キーと値を反復処理し、mean()メソッドを使ってそれぞれの平均値を計算します。

• 最終的な結果をコンソールに表示します。

defaultdictを使ったより簡潔な方法

collectionsモジュールのdefaultdictを利用すると、キーの存在チェックを省略でき、コードをさらに簡潔にできます。

from statistics import mean
from collections import defaultdict

my_list = [{'hi' : 24, 'there' : 81, 'how' : 11},
   {'hi' : 16, 'how' : 78, 'doing' : 63}]

my_result = defaultdict(list)
for sub in my_list:
   for key, val in sub.items():
      my_result[key].append(val)

my_result = {key: mean(vals) for key, vals in my_result.items()}

print("The result is : ")
print(my_result)

この方法では、存在しないキーにアクセスした際に自動的に空のリストが生成されるため、条件分岐が不要になります。また、辞書内包表記と組み合わせることで、平均値への変換も1行で記述できます。データ量が多い場合や可読性を重視したい場合には、こちらのアプローチがおすすめです。