python - 根据条件更改数据框中的值

Question

这是我的数据框:

import pandas as pd
list = ['RK','G', 'CUSTOM', 'CUSTOM', 'KL', 'kj']
df=pd.DataFrame(list,columns=['A'])

我想在这里做的是找出 A 列中等于 CUSTOM 的所有条目，并将其替换为 CUSTOM1 和 CUSTOM2。

输出应该是:

       A
0      RK
1      G
2    CUSTOM1
3    CUSTOM2
4     KL
5     kj

如果有 3 个 CUSTOM 而不是 2 个，输出应该是 CUSTOM1、CUSTOM2 和 CUSTOM3。

谢谢。

Answer 1

使用cumsum + np.where -

m = df.A.eq('CUSTOM')
df.A = np.where(m, df.A + m.cumsum().astype(str), df.A)

df
         A
0       RK
1        G
2  CUSTOM1
3  CUSTOM2
4       KL
5       kj

---

使用 pd.Series.where/mask 的类似解决方案 -

df.A = df.A.where(~m, df.A + m.cumsum().astype(str))

或者，

df.A = df.A.mask(m, df.A + m.cumsum().astype(str))

df
         A
0       RK
1        G
2  CUSTOM1
3  CUSTOM2
4       KL
5       kj