groupby操作中的熊猫索引

Question

我正在尝试将 groupby 对象中每个单独记录的索引（或运行计数）放入列中。我不必是 groupby，但顺序必须保持不变，例如，我想按 C 列排序和重新索引：

df = pd.DataFrame([[1, 2, 'Foo'],

                   [1, 3, 'Foo'],

                   [4, 6,'Bar'],

                   [7,8,'Bar']],

                  columns=['A', 'B', 'C'])

Out[72]: 

   A  B    C

0  1  2  Foo

1  1  3  Foo

2  4  6  Bar

3  7  8  Bar

我想要的输出是：

Out[75]: 

   A  B    C  sorted

0  1  2  Foo       1

1  1  3  Foo       2

2  4  6  Bar       1

3  7  8  Bar       2

看起来这应该很容易，但是如果不循环遍历整个数据框，我尝试过的任何事情都没有真正接近，我希望避免这种情况。谢谢

Answer 1

尝试cumcount：>>> df = pd.DataFrame([[1, 2, 'Foo'],...                    [1, 3, 'Foo'],...                    [4, 6,'Bar'],...                    [7,8,'Bar']],...                   columns=['A', 'B', 'C'])>>> df["sorted"]=df.groupby("C").cumcount()+1>>> df   A  B    C  sorted0  1  2  Foo       11  1  3  Foo       22  4  6  Bar       13  7  8  Bar       2