第二列中值的条件累计计数

wtzytmuj 于 2022-09-18 发布在 Java

关注(0)|答案(1)|浏览(135)

我想根据列KEY中的值在列flag中填写数字。

如果列KEY中的值保持不变，我希望每两行填充相同的数字，而不是使用cumcount()填充递增的数字。
如果KEY列中的值更改，则填充的数字也会更改。

下面是示例，df1是我想要的df0。

df0 = pd.DataFrame({'KEY':['0','0','0','0','1','1','1','2','2','2','2','2','3','3','3','3','3','3','4','5','6']})

df1 = pd.DataFrame({'KEY':['0','0','0','0','1','1','1','2','2','2','2','2','3','3','3','3','3','3','4','5','6'],
                    'flag':['0','0','1','1','2','2','3','4','4','5','5','6','7','7','8','8','9','9','10','11','12']})

python

来源：https://stackoverflow.com/questions/64545116/conditional-cumcount-of-values-in-second-column

1条答案

按热度按时间

fykwrbwg1#

You want to get the cumcount and add one. Then use %2 to differentiate between odd or even rows. Then, take the cumulative sum and subtract 1 to start counting from zero.

You can use:

df0['flag'] = ((df0.groupby('KEY').cumcount() + 1) % 2).cumsum() - 1
df0
Out[1]: 
   KEY  flag
0    0      0
1    0      0
2    0      1
3    0      1
4    1      2
5    1      2
6    1      3
7    2      4
8    2      4
9    2      5
10   2      5
11   2      6
12   3      7
13   3      7
14   3      8
15   3      8
16   3      9
17   3      9
18   4     10
19   5     11
20   6     12

赞(0）回复(0）举报 2022-09-18

我来回答

第二列中值的条件累计计数

1条答案

相关问题

热门标签

最新问答