przekształcić wierszy do kolumn w Pandy dataframe

Question

W pand jak przejść od:

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 

do B:

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

Próbowałem rozwiązań w innych odpowiedzi, na przykład here i here, ale nikt nie wydawał się robić tego, co chcę.

Zasadniczo chcę zamienić wiersze kolumnami i upuścić indeks, ale jak to zrobić?

Answer 1

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

Answer 2

To jest moje rozwiązanie

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