1. ホーム
  2. python

[解決済み] Python Pandas : group byとaverageでグループ分け?

2022-09-12 02:57:48

質問

このようなデータフレームがあります。

cluster  org      time
   1      a       8
   1      a       6
   2      h       34
   1      c       23
   2      d       74
   3      w       6 

クラスタごとに1組織あたりの平均時間を計算したいのですが。

期待される結果

cluster mean(time)
1       15 ((8+6)/2+23)/2
2       54   (74+34)/2
3       6

Pandasでどのように行うかわかりません、誰か助けてください。

どのように解決するのですか?

の組み合わせで平均を取りたい場合、まず最初に ['cluster', 'org'] の組み合わせで平均をとり、次に cluster のグループであれば、使えます。

In [59]: (df.groupby(['cluster', 'org'], as_index=False).mean()
            .groupby('cluster')['time'].mean())
Out[59]:
cluster
1          15
2          54
3           6
Name: time, dtype: int64

の平均が欲しい場合は cluster のグループのみの平均が欲しい場合は、以下のようになります。

In [58]: df.groupby(['cluster']).mean()
Out[58]:
              time
cluster
1        12.333333
2        54.000000
3         6.000000

また groupby['cluster', 'org'] を使用し、その後に mean() :

In [57]: df.groupby(['cluster', 'org']).mean()
Out[57]:
               time
cluster org
1       a    438886
        c        23
2       d      9874
        h        34
3       w         6