팬더에 의해 행 나는이 데이터 집합을 변환하기 위해 노력하고있어 열

Question

에 그룹을 결합

A B C1 C2 C3 
1 x1 a a b 
2 x2 b a null 

df = pd.DataFrame({ 'A': [1, 1, 1, 2, 2], 
       'B': ['x1', 'x1', 'x1', 'x2', 'x2'], 
       'C': ['a', 'a', 'b', 'b', 'a'] 
       }) 

여기에서 대답은 어떻게 든 가까운하지만 피벗은 잘 작동을하지 않습니다 나를 위해. How to do a transpose a dataframe group by key on pandas?

Answer 1

사용 groupby + apply -

v = df.groupby(['A' ,'B']).C.apply(lambda x: x.tolist()) 

df = pd.DataFrame(v.tolist(), index=v.index)\ 
     .rename(columns=lambda x: x + 1)\ 
     .add_prefix('C')\ 
     .reset_index() 
df 

    A B C1 C2 C3 
0 1 x1 a a  b 
1 2 x2 b a None 

Answer 2

둘 다 제안이 작동하지만 일을 받아 들일 수, set_index 및

In [196]: (df.set_index(['A', 'B', df.groupby(['A', 'B']).cumcount()+1])['C'] 
      .unstack() 
      .add_prefix('C') 
      .reset_index()) 
Out[196]: 
    A B C1 C2 C3 
0 1 x1 a a  b 
1 2 x2 b a None