Python Pandas: dikelompokkan menurut kelompok menurut dan rata-rata?

93

Saya memiliki dataframe seperti ini:

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

Saya ingin menghitung rata-rata waktu per org per cluster.

Hasil yang diharapkan:

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

Saya tidak tahu bagaimana melakukannya di Pandas, ada yang bisa membantu?

UserYmY
sumber
2
Maaf yang Anda ingin df.groupby(['org','cluster']).mean()? ini tidak terlalu berarti untuk dataset Anda dibandingkan dengan ini:df.groupby(['cluster','org']).mean()
EdChum
@EdChum terima kasih atas bantuan Anda, tetapi saya ingin tidak satupun dari mereka. Tapi "rata-rata waktu rata-rata per org". Silakan lihat hasil yang diharapkan (diedit)
UserYmY

Jawaban:

133

Jika Anda ingin pertama-tama mengambil jahat pada kombinasi ['cluster', 'org']dan kemudian mengambil jahat pada clusterkelompok, Anda dapat menggunakan:

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

Jika Anda hanya menginginkan mean dari clustergrup, maka Anda dapat menggunakan:

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

Anda juga dapat menggunakan groupbypada ['cluster', 'org']dan kemudian menggunakan 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
Nol
sumber
Tapi saya ingin satu nomor per cluster (rata-rata waktu rata-rata per org). Jadi hasilnya hanya cluster dan waktu rata-rata
UserYmY
1
Yang pertama adalah yang saya butuhkan, terima kasih - apakah ada cara mudah untuk menambahkan kolom di df yang berisi nilai rata-rata ini untuk cluster baris?
Mr_and_Mrs_D
12

Saya hanya akan melakukan ini, yang secara harfiah mengikuti logika yang Anda inginkan:

df.groupby(['org']).mean().groupby(['cluster']).mean()
Vince Payandeh
sumber