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


93

У меня есть такой фрейм данных:

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

Не знаю, как это сделать в Пандах, может кто поможет?


2
Извините, вы хотите df.groupby(['org','cluster']).mean()? это не так важно для вашего набора данных в отличие от этого:df.groupby(['cluster','org']).mean()
EdChum

@EdChum спасибо за вашу помощь, но я не хочу никого из них. Но «среднее среднее время на организацию». Пожалуйста, посмотрите ожидаемый результат (отредактированный)
UserYmY

Ответы:


133

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

Но мне нужно одно число на кластер (среднее среднее время на организацию). Таким образом, результат - только кластер и среднее время
UserYmY

1
Первое - это то, что мне нужно, спасибо - есть ли простой способ добавить столбец в df, содержащий эти средние значения для кластера строк?
Mr_and_Mrs_D

12

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

df.groupby(['org']).mean().groupby(['cluster']).mean()
Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.