Python Pandas: จัดกลุ่มตามกลุ่มและค่าเฉลี่ย?


93

ฉันมี dataframe แบบนี้:

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()
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.