pandas groupby เรียงลำดับภายในกลุ่ม


166

ฉันต้องการจัดกลุ่ม dataframe ของฉันด้วยสองคอลัมน์จากนั้นเรียงลำดับผลลัพธ์รวมภายในกลุ่ม

In [167]:
df

Out[167]:
count   job source
0   2   sales   A
1   4   sales   B
2   6   sales   C
3   3   sales   D
4   7   sales   E
5   5   market  A
6   3   market  B
7   2   market  C
8   4   market  D
9   1   market  E

In [168]:
df.groupby(['job','source']).agg({'count':sum})

Out[168]:
            count
job     source  
market  A   5
        B   3
        C   2
        D   4
        E   1
sales   A   2
        B   4
        C   6
        D   3
        E   7

ตอนนี้ฉันต้องการเรียงลำดับคอลัมน์นับตามลำดับจากมากไปน้อยภายในแต่ละกลุ่ม จากนั้นใช้เฉพาะสามแถวบนสุด ที่จะได้รับสิ่งที่ชอบ:

            count
job     source  
market  A   5
        D   4
        B   3
sales   E   7
        C   6
        B   4

คำตอบ:


147

สิ่งที่คุณต้องการทำจริง ๆ แล้วเป็น groupby อีกครั้ง (จากผลลัพธ์ของ groupby แรก): เรียงลำดับและรับองค์ประกอบสามอย่างแรกต่อกลุ่ม

เริ่มต้นจากผลลัพธ์ของ groupby แรก:

In [60]: df_agg = df.groupby(['job','source']).agg({'count':sum})

เราจัดกลุ่มตามระดับแรกของดัชนี:

In [63]: g = df_agg['count'].groupby(level=0, group_keys=False)

จากนั้นเราต้องการเรียงลำดับ ('คำสั่ง') แต่ละกลุ่มและนำองค์ประกอบสามข้อแรก:

In [64]: res = g.apply(lambda x: x.order(ascending=False).head(3))

อย่างไรก็ตามสำหรับสิ่งนี้มีฟังก์ชั่นทางลัดในการทำสิ่งนี้nlargest:

In [65]: g.nlargest(3)
Out[65]:
job     source
market  A         5
        D         4
        B         3
sales   E         7
        C         6
        B         4
dtype: int64

จะมีวิธีสรุปรวมทุกสิ่งที่ไม่อยู่ในผลลัพธ์สามอันดับแรกต่อกลุ่มและเพิ่มลงในกลุ่มแหล่งที่มาที่เรียกว่า "อื่น ๆ " สำหรับแต่ละงานหรือไม่
JoeDanger

30
orderเลิกใช้แล้วsort_valuesแทน
zthomas.nc

ขอบคุณสำหรับคำตอบที่ดี สำหรับขั้นตอนต่อไปจะมีวิธีกำหนดลำดับการเรียงตามค่าในคอลัมน์ groupby หรือไม่ ตัวอย่างเช่นเรียงจากน้อยไปมากหากค่าคือ 'ซื้อ' และเรียงจากมากไปหาน้อยหากค่าคือ 'ขาย'
Bowen Liu

173

คุณสามารถทำได้เพียงครั้งเดียวโดยใช้การเรียงลำดับแรกและใช้หัวเพื่อเลือก 3 อันดับแรกของแต่ละกลุ่ม

In[34]: df.sort_values(['job','count'],ascending=False).groupby('job').head(3)

Out[35]: 
   count     job source
4      7   sales      E
2      6   sales      C
1      4   sales      B
5      5  market      A
8      4  market      D
6      3  market      B

14
ไม่groupbyรับประกันว่าคำสั่งซื้อจะถูกเก็บไว้?
toto_tico

52
ดูเหมือนว่ามันจะ; จากเอกสารของ groupby : groupby รักษาลำดับของแถวภายในแต่ละกลุ่ม
toto_tico

10
toto_tico- ถูกต้องอย่างไรก็ตามต้องระมัดระวังในการตีความคำสั่งนั้น ลำดับของแถวภายในกลุ่มเดียวจะถูกเก็บไว้อย่างไรก็ตาม groupby มีคำสั่ง sort = True ตามค่าเริ่มต้นซึ่งหมายความว่ากลุ่มเหล่านั้นอาจถูกเรียงลำดับบนคีย์ กล่าวอีกนัยหนึ่งถ้าไฟล์ข้อมูลของฉันมีคีย์ (ที่อินพุต) 3 2 2 1 .. กลุ่มโดยวัตถุจะแสดง 3 กลุ่มตามลำดับ 1 2 3 (เรียงลำดับ) ใช้ sort = False เพื่อให้แน่ใจว่าคำสั่งกลุ่มและลำดับแถวนั้นถูกสงวนไว้
user2103050

4
หัว (3) ให้ผลลัพธ์มากกว่า 3 รายการ?
Nabin

27

นี่คือตัวอย่างอื่น ๆ ของการรับ 3 อันดับแรกในการเรียงลำดับและการเรียงลำดับภายในกลุ่ม:

In [43]: import pandas as pd                                                                                                                                                       

In [44]:  df = pd.DataFrame({"name":["Foo", "Foo", "Baar", "Foo", "Baar", "Foo", "Baar", "Baar"], "count_1":[5,10,12,15,20,25,30,35], "count_2" :[100,150,100,25,250,300,400,500]})

In [45]: df                                                                                                                                                                        
Out[45]: 
   count_1  count_2  name
0        5      100   Foo
1       10      150   Foo
2       12      100  Baar
3       15       25   Foo
4       20      250  Baar
5       25      300   Foo
6       30      400  Baar
7       35      500  Baar


### Top 3 on sorted order:
In [46]: df.groupby(["name"])["count_1"].nlargest(3)                                                                                                                               
Out[46]: 
name   
Baar  7    35
      6    30
      4    20
Foo   5    25
      3    15
      1    10
dtype: int64


### Sorting within groups based on column "count_1":
In [48]: df.groupby(["name"]).apply(lambda x: x.sort_values(["count_1"], ascending = False)).reset_index(drop=True)
Out[48]: 
   count_1  count_2  name
0       35      500  Baar
1       30      400  Baar
2       20      250  Baar
3       12      100  Baar
4       25      300   Foo
5       15       25   Foo
6       10      150   Foo
7        5      100   Foo

9

ลองใช้วิธีนี้แทน

วิธีง่ายๆในการทำ 'groupby' และเรียงลำดับจากมากไปน้อย

df.groupby(['companyName'])['overallRating'].sum().sort_values(ascending=False).head(20)

8

หากคุณไม่ต้องการรวมคอลัมน์ให้ใช้คำตอบของ @ tvashtar หากคุณต้องการหาผลรวมคุณสามารถใช้ @joris 'answer หรืออันนี้ซึ่งคล้ายกันมาก

df.groupby(['job']).apply(lambda x: (x.groupby('source')
                                      .sum()
                                      .sort_values('count', ascending=False))
                                     .head(3))
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.