Pandas DataFrame Group โดยสองคอลัมน์และรับการนับ


166

ฉันมี dataframe นุ่นในรูปแบบต่อไปนี้:

df = pd.DataFrame([[1.1, 1.1, 1.1, 2.6, 2.5, 3.4,2.6,2.6,3.4,3.4,2.6,1.1,1.1,3.3], list('AAABBBBABCBDDD'), [1.1, 1.7, 2.5, 2.6, 3.3, 3.8,4.0,4.2,4.3,4.5,4.6,4.7,4.7,4.8], ['x/y/z','x/y','x/y/z/n','x/u','x','x/u/v','x/y/z','x','x/u/v/b','-','x/y','x/y/z','x','x/u/v/w'],['1','3','3','2','4','2','5','3','6','3','5','1','1','1']]).T
df.columns = ['col1','col2','col3','col4','col5']

DF:

   col1 col2 col3     col4 col5
0   1.1    A  1.1    x/y/z    1
1   1.1    A  1.7      x/y    3
2   1.1    A  2.5  x/y/z/n    3
3   2.6    B  2.6      x/u    2
4   2.5    B  3.3        x    4
5   3.4    B  3.8    x/u/v    2
6   2.6    B    4    x/y/z    5
7   2.6    A  4.2        x    3
8   3.4    B  4.3  x/u/v/b    6
9   3.4    C  4.5        -    3
10  2.6    B  4.6      x/y    5
11  1.1    D  4.7    x/y/z    1
12  1.1    D  4.7        x    1
13  3.3    D  4.8  x/u/v/w    1

ตอนนี้ฉันต้องการจัดกลุ่มนี้ด้วยสองคอลัมน์ดังนี้:

df.groupby(['col5','col2']).reset_index()

เอาท์พุท:

             index col1 col2 col3     col4 col5
col5 col2                                      
1    A    0      0  1.1    A  1.1    x/y/z    1
     D    0     11  1.1    D  4.7    x/y/z    1
          1     12  1.1    D  4.7        x    1
          2     13  3.3    D  4.8  x/u/v/w    1
2    B    0      3  2.6    B  2.6      x/u    2
          1      5  3.4    B  3.8    x/u/v    2
3    A    0      1  1.1    A  1.7      x/y    3
          1      2  1.1    A  2.5  x/y/z/n    3
          2      7  2.6    A  4.2        x    3
     C    0      9  3.4    C  4.5        -    3
4    B    0      4  2.5    B  3.3        x    4
5    B    0      6  2.6    B    4    x/y/z    5
          1     10  2.6    B  4.6      x/y    5
6    B    0      8  3.4    B  4.3  x/u/v/b    6

ฉันต้องการได้รับการนับโดยแต่ละแถวดังนี้ ผลลัพธ์ที่คาดหวัง:

col5 col2 count
1    A      1
     D      3
2    B      2
etc...

ฉันจะได้รับผลลัพธ์ที่คาดหวังได้อย่างไร และฉันต้องการค้นหาจำนวนที่มากที่สุดสำหรับค่า 'col2' แต่ละอัน?


คำถามที่คล้ายกันมากเพียงแค่ขึ้นมาเมื่อวานนี้ .. เห็นที่นี่
bdiamante

หมายเหตุเกี่ยวกับประสิทธิภาพรวมถึงทางเลือก: Pandas groupby.size vs series.value_counts vs ชุดสะสมพบกับหลายซีรี่ส์
jpp

คำตอบ:


116

ตามด้วยคำตอบของ @ Andy คุณสามารถทำตามเพื่อแก้ไขคำถามที่สองของคุณ:

In [56]: df.groupby(['col5','col2']).size().reset_index().groupby('col2')[[0]].max()
Out[56]: 
      0
col2   
A     3
B     2
C     1
D     3

1
ฉันสามารถรับค่า "col5" สำหรับสิ่งนี้เช่น C ... 1 ... 3 ได้ไหม
Nilani Algiriyage

141

คุณกำลังมองหาsize:

In [11]: df.groupby(['col5', 'col2']).size()
Out[11]:
col5  col2
1     A       1
      D       3
2     B       2
3     A       3
      C       1
4     B       1
5     B       2
6     B       1
dtype: int64

เพื่อให้ได้คำตอบเดียวกับ waitkuo ("คำถามที่สอง") แต่สะอาดกว่าเล็กน้อยคือจัดกลุ่มตามระดับ:

In [12]: df.groupby(['col5', 'col2']).size().groupby(level=1).max()
Out[12]:
col2
A       3
B       2
C       1
D       3
dtype: int64

1
ฉันไม่รู้ว่าทำไมฉันลืมสิ่งนี้: O, แล้วคำถามที่สองของฉันจะหาค่าที่มากที่สุดสำหรับค่า "col2" แต่ละค่าและรับค่า "col5" ที่สอดคล้องกันได้อย่างไร
Nilani Algiriyage

23

การใส่ข้อมูลที่เป็น dataframe หมีแพนด้าและให้ชื่อคอลัมน์

import pandas as pd
df = pd.DataFrame([['A','C','A','B','C','A','B','B','A','A'], ['ONE','TWO','ONE','ONE','ONE','TWO','ONE','TWO','ONE','THREE']]).T
df.columns = [['Alphabet','Words']]
print(df)   #printing dataframe.

นี่คือข้อมูลที่พิมพ์ของเรา:

ป้อนคำอธิบายรูปภาพที่นี่

สำหรับการทำกลุ่ม dataframe ในหมีแพนด้าและเคาน์เตอร์ ,
คุณจำเป็นต้องให้หนึ่งคอลัมน์อื่น ๆ ซึ่งนับการจัดกลุ่มที่ให้โทรคอลัมน์เป็น"เคาน์เตอร์"ใน dataframe

แบบนี้:

df['COUNTER'] =1       #initially, set that counter to 1.
group_data = df.groupby(['Alphabet','Words'])['COUNTER'].sum() #sum function
print(group_data)

เอาท์พุท:

ป้อนคำอธิบายรูปภาพที่นี่


9
ฉันจะทำให้คอลัมน์ตัวอักษร (เช่น. A) ทำซ้ำด้านล่างและไม่เว้นช่องว่างในคอลัมน์แรกได้อย่างไร
blissweb

วิธีการเข้าถึงคุณค่าของแต่ละกลุ่มซึ่งเป็นผลรวมตามตัวอักษรและคำ?
ราหุล Goyal

21

วิธีการแก้ปัญหาสำนวนที่ใช้เพียงกลุ่มเดียวโดย

(df.groupby(['col5', 'col2']).size() 
   .sort_values(ascending=False) 
   .reset_index(name='count') 
   .drop_duplicates(subset='col2'))

  col5 col2  count
0    3    A      3
1    1    D      3
2    5    B      2
6    3    C      1

คำอธิบาย

ผลลัพธ์ของsizeเมธอดgroupby คือ Series ที่มีcol5และcol2อยู่ในดัชนี จากที่นี่คุณสามารถใช้วิธีการ groupby อื่นเพื่อค้นหาค่าสูงสุดของแต่ละค่าใน col2แต่ไม่จำเป็นต้องทำ คุณสามารถเรียงลำดับค่าทั้งหมดจากมากไปน้อยแล้วเก็บเฉพาะแถวที่มีการเกิดขึ้นครั้งแรกcol2ด้วยdrop_duplicatesวิธีการ


ไม่มีพารามิเตอร์ที่ถูกเรียกnameในreset_index()pandas รุ่นปัจจุบัน: pandas.pydata.org/pandas-docs/stable/generated/…
mmBs


ตกลงฉันไม่ดี ฉันใช้มันเมื่อทำงานกับไม่ได้DataFrame Seriesขอบคุณสำหรับลิงค์
mmBs

2

หากคุณต้องการเพิ่มคอลัมน์ใหม่ (พูดว่า 'count_column') ที่มีการนับกลุ่ม 'ลงใน dataframe:

df.count_column=df.groupby(['col5','col2']).col5.transform('count')

(ฉันเลือก 'col5' เนื่องจากไม่มี Nan)


-2

คุณสามารถใช้ฟังก์ชันในตัวนับตามด้วยฟังก์ชัน groupby

df.groupby(['col5','col2']).count()
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.