Python Pandas: группировать по группам и средним?

У меня есть df, как это:

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

Я хотел бы рассчитать среднее время на один узел на кластер.

Ожидаемый результат:

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

Я не знаю, как это сделать в Pandas, может ли кто-нибудь помочь?

Ответы

Ответ 1

Если вы хотите сначала принять значение в комбинации ['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

Ответ 2

Я бы просто сделал это, что буквально следует за вашей желаемой логикой:

df.groupby(['org']).mean().groupby(['cluster']).mean()