循环遍历熊猫表，按条件更改其他列的值

Question

我目前正在尝试根据另一个列值更改熊猫列的值。到目前为止没问题，但我想遍历两列以设置我的表，如下例所示：

我有一个这样的熊猫表：

id  id2
1    0
1    0
1    0
2    0
2    0
2    0
...

我想得到这样的东西：

id  id2
1    1
1    2
1    3
2    1
2    2
2    3
...

所以我必须以某种方式检查id 列的值，并根据它更改id2 的值。到目前为止，我只有那个：

n=1
m=50
df.loc[df.id==n, 'id2'] = m

这给了我以下输出，但不是我想要做的：

 id  id2
 1   50
 1   50
 1   50
 2   50
 2   50
 2   50

我是编程新手，我尝试的所有循环都失败了。如果有人能解释我该怎么做，我将非常感激！

Answer 1

也许DataFrame.rank() 方法是你需要的。

例子：

import pandas as pd
df = pd.DataFrame({'id1':[1,1,1,2,2,2,3,3],'id2':[0,0,0,0,0,0,0,0]})
df['dummy_rank'] = df['id2'].groupby(df['id1']).rank(method='first').apply(int)

关键是“方法”参数，您可以通过文档熊猫了解更多信息。

Answer 2

将 cumsum() 与 groupby() 一起使用

df = pd.DataFrame({'id':[1,1,1,2,2,2], 'id2':[0,0,0,0,0,0]})

df['id2'] =1
df['id2'] = df.groupby('id')['id2'].cumsum()
print(df)

输出：

   id  id2
0   1    1
1   1    2
2   1    3
3   2    1
4   2    2
5   2    3