Я хочу напечатать результат группировки с помощью Pandas.
У меня есть dataframe:
import pandas as pd
df = pd.DataFrame({'A': ['one', 'one', 'two', 'three', 'three', 'one'], 'B': range(6)})
print df
A B
0 one 0
1 one 1
2 two 2
3 three 3
4 three 4
5 one 5
При печати после группировки "A" у меня есть следующее:
print df.groupby('A')
<pandas.core.groupby.DataFrameGroupBy object at 0x05416E90>
Как я могу напечатать группу данных?
Если я это сделаю:
print df.groupby('A').head()
Я получаю dataframe, как если бы он не был сгруппирован:
A B
A
one 0 one 0
1 one 1
two 2 two 2
three 3 three 3
4 three 4
one 5 one 5
Я ожидал чего-то вроде:
A B
A
one 0 one 0
1 one 1
5 one 5
two 2 two 2
three 3 three 3
4 three 4