python - pandas 中的分组行

Question

我有一个如下所示的数据框:

In [4]:
import pandas as pd
df = pd.DataFrame( {'a':['A','A','B','B','B','C'], 'b':[1,2,5,5,4,6]})
df
Out[4]:
    a   b
0   A   1
1   A   2
2   B   5
3   B   5
4   B   4
5   C   6

我只想将a 列中具有相同值的行分组。期望的输出是这样的:

    df
    Out[4]:
        a   b
    0   A   1
            2
    1   B   5
            5
            4
    2   C   6

编辑:

抱歉，实际上想要的输出可能是这样的:

    df
    Out[4]:
        b
    A   1
        2
    B   5
        5
        4
    C   6

Answer 1

我认为您正在寻找 set_index而不是groupby :

In [11]: df.set_index('a')
Out[11]:
   b
a
A  1
A  2
B  5
B  5
B  4
C  6