Reshape hàng để cột trong gấu trúc dataframe

Question

Trong gấu trúc thế nào để đi từ một:

a = pd.DataFrame({'foo': ['m', 'm', 'm', 's', 's', 's'], 
        'bar': [1, 2, 3, 4, 5, 6]}) 
>>> a 
    bar foo 
0 1 m 
1 2 m 
2 3 m 
3 4 s 
4 5 s 
5 6 s 

để b:

b = pd.DataFrame({'m': [1, 2, 3], 
        's': [4, 5, 6]}) 
>>> b 
    m s 
0 1 4 
1 2 5 
2 3 6 

tôi đã cố gắng giải pháp trong câu trả lời khác, ví dụ here và here nhưng không ai có vẻ làm những gì tôi muốn.

Về cơ bản, tôi muốn trao đổi các hàng có cột và thả chỉ mục, nhưng cách thực hiện?

Answer 1

a.set_index(
    [a.groupby('foo').cumcount(), 'foo'] 
).bar.unstack() 

Answer 2

Đây là giải pháp của tôi

a = pd.DataFrame({'foo': ['m', 'm', 'm', 's', 's', 's'], 
        'bar': [1, 2, 3, 4, 5, 6]}) 
a.pivot(columns='foo', values='bar').apply(lambda x: pd.Series(x.dropna().values)) 

foo m s 
0 1.0 4.0 
1 2.0 5.0 
2 3.0 6.0