2016-10-03 7 views

답변

0

사용자 정의 기능에서 선택 열 ab을 선택한다고 생각하십니까?

df = pd.DataFrame({'a':[1,2,3], 
        'b':[4,5,6], 
        'c':[7,8,9], 
        'D':[1,3,5], 
        'E':[5,3,6]}) 

print (df) 
    D E a b c 
0 1 5 1 4 7 
1 3 3 2 5 8 
2 5 6 3 6 9 


def f(x): 
# print (x) 
    print (x[['a','b']]) 
    return x[['a','b']] 

print (df.groupby(['a','b','c']).apply(f)) 

    a b 
0 1 4 
    a b 
0 1 4 
    a b 
1 2 5 
    a b 
2 3 6 
    a b 


    a b 
0 1 4 
1 2 5 
2 3 6 

또는 어쩌면 당신은 filter이 필요합니다

print (df.groupby(['a','b','c']).filter(lambda x: (x.b == 4) & (x.c == 7))) 
    D E a b c 
0 1 5 1 4 7 
+0

내가 질문을 이해한다면 확실하지 않다. 더 설명해 주시겠습니까? – jezrael

0
[name for name,group in df.groupby(['a', 'b', 'c'])] 
관련 문제