使用groupby的熊猫标准化

Question

我有两个系列的专栏第一列日期范围为2015-01-01至2019-01-01，第二列具有一些随机值，我想创建一个新列，其外观应如下所示

我有如下的熊猫专栏

A1             B1

2015-01-01     A
2015-02-01     A
2015-03-01     A
2015-04-01     A
2015-01-01     B
2015-02-01.    B
-----

并且我想要下面的新列

A1              B1    B

2015-01-01       A   0
2015-02-01       A   1
2015-03-01       A   2
2015-05-01.      A   3
2015-01-01       B   0
2015-02-01.      B   1

我认为我应该在B1上使用groupby函数，但不确定如何做到这一点

Answer 1

groupby.cumcount

groupby.cumcount

就位

df.assign(B=df.groupby('B1').cumcount())

            A1 B1  B
0   2015-01-01  A  0
1   2015-02-01  A  1
2   2015-03-01  A  2
3   2015-04-01  A  3
4   2015-01-01  B  0
5   2015-02-01  B  1