pandas 그룹 by 그룹의 최대 값 이 있 는 줄 가 져 오기

전재 하 다http://www.guoguoday.com/post/pandas그룹 by 그룹의 최대 값 이 있 는 줄 가 져 오기 /
pandas 그룹 by 그룹의 최대 값 이 있 는 줄 가 져 오기
10/May 2016
python  pandas
pandas 그룹 by 그룹의 최대 값 이 있 는 줄 가 져 오기
아래 의 DataFrame 와 같이 Mt 로 나 누 어 Count 가 가장 큰 줄 을 꺼 냅 니 다.

import pandas as pd
df = pd.DataFrame({'Sp':['a','b','c','d','e','f'], 'Mt':['s1', 's1', 's2','s2','s2','s3'], 'Value':[1,2,3,4,5,6], 'Count':[3,2,5,10,10,6]})
df

 
Count
Mt
Sp
Value
0
3
s1
a
1
1
2
s1
b
2
2
5
s2
c
3
3
10
s2
d
4
4
10
s2
e
5
5
6
s3
f
6
방법 1: 그룹 에서 Count 의 가장 큰 줄 을 걸 러 냅 니 다.

df.groupby('Mt').apply(lambda t: t[t.Count==t.Count.max()])

 
 
Count
Mt
Sp
Value
Mt
 
 
 
 
 
s1
0
3
s1
a
1
s2
3
10
s2
d
4
4
10
s2
e
5
s3
5
6
s3
f
6
방법 2: transform 으로 원 dataframe 의 index 를 가 져 온 다음 필요 한 줄 을 걸 러 냅 니 다.

print df.groupby(['Mt'])['Count'].agg(max)

idx=df.groupby(['Mt'])['Count'].transform(max)
print idx
idx1 =  idx == df['Count']
print idx1

df[idx1]

Mt
s1     3
s2    10
s3     6
Name: Count, dtype: int64
0     3
1     3
2    10
3    10
4    10
5     6
dtype: int64
0     True
1    False
2    False
3     True
4     True
5     True
dtype: bool

 
Count
Mt
Sp
Value
0
3
s1
a
1
3
10
s2
d
4
4
10
s2
e
5
5
6
s3
f
6
위의 방법 중 하 나 는 3, 4 줄 의 값 이 모두 최대 치 라 는 것 이다. 이렇게 여러 줄 로 되 돌아 갔다. 만약 한 줄 만 되 돌아 가면?
방법 3: idmax (구 버 전 pandas 는 argmax)

idx = df.groupby('Mt')['Count'].idxmax()
print idx

df.iloc[idx]

Mt
s1    0
s2    3
s3    5
Name: Count, dtype: int64

 
Count
Mt
Sp
Value
0
3
s1
a
1
3
10
s2
d
4
5
6
s3
f
6

df.iloc[df.groupby(['Mt']).apply(lambda x: x['Count'].idxmax())]

 
Count
Mt
Sp
Value
0
3
s1
a
1
3
10
s2
d
4
5
6
s3
f
6

def using_apply(df):
    return (df.groupby('Mt').apply(lambda subf: subf['Value'][subf['Count'].idxmax()]))

def using_idxmax_loc(df):
    idx = df.groupby('Mt')['Count'].idxmax()
    return df.loc[idx, ['Mt', 'Value']]

print using_apply(df)

using_idxmax_loc(df)

Mt
s1    1
s2    4
s3    6
dtype: int64

 
Mt
Value
0
s1
1
3
s2
4
5
s3
6
방법 4: 먼저 순 서 를 정 한 다음 에 각 조 에서 첫 번 째 를 취한 다.

df.sort('Count', ascending=False).groupby('Mt', as_index=False).first()

 
Mt
Count
Sp
Value
0
s1
3
a
1
1
s2
10
d
4
2
s3
6
f
6
그 문 제 는 또 왔 다. 만약 에 가장 큰 값 이 있 는 줄 을 꺼 내 려 고 하지 않 았 다 면, 예 를 들 어 중간 값 이 있 는 줄 은?
사고방식 은 여전히 유사 하 다. 구체 적 인 문법 에 있어 서 수정 을 해 야 할 수도 있다. 예 를 들 어 방법 1 과 2 는 max 알고리즘 을 수정 하고 방법 3 은 스스로 index 로 돌아 가 는 방법 을 실현 해 야 한다.어쨌든 groupby 이후 각 그룹 은 하나의 dataframe 입 니 다.