groupby操作中的熊猫索引

krflol 发表于 Dev

我试图将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

看来这确实很容易，但是我尝试过的所有事情都不会在没有遍历整个数据帧的情况下真正接近，这是我希望避免的。谢谢

格热哥兹·斯基宾斯基

尝试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  sorted
0  1  2  Foo       1
1  1  3  Foo       2
2  4  6  Bar       1
3  7  8  Bar       2

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。