使用 Pandas/Python 为列中的重复项生成唯一值

如何解决使用 Pandas/Python 为列中的重复项生成唯一值

我有一个数据集 df，我想在其中通过在末尾放置数字来为类型列中的值创建唯一 ID。

数据

type    total   free  use
a       10      5     5
a       10      4     6
a       10      1     9
a       10      8     2
a       10      3     7
b       20      5     5
b       20      3     7
b       20      2     8
b       20      6     4
b       20      2     8

需要

type    total   free  use
a       10      5     5
a1      10      4     6
a2      10      1     9
a3      10      8     2
a4      10      3     7
b       20      5     5
b1      20      3     7
b2      20      2     8
b3      20      6     4
b4      20      2     8

做

我可以通过这样做在 R 中做到这一点，但不确定如何在 Python 中做到这一点：

library(data.table)
setDT(DT)

DT[,run_id := rleid(ID)]
DT[DT[,.SD[1L],by = run_id][duplicated(ID),ID := paste0('list',.I)],on = 'run_id',ID := i.ID][]

我正在研究这个，感谢任何输入

解决方法

您可以使用groupby.cumcount：

df['type'] += np.where(df['type'].duplicated(),df.groupby('type').cumcount().astype(str),'')

或类似地使用 loc 更新：

df.loc[df['type'].duplicated(),'type'] += df.groupby('type').cumcount().astype(str)

输出：

  type  total  free  use
0    a     10     5    5
1   a1     10     4    6
2   a2     10     1    9
3   a3     10     8    2
4   a4     10     3    7
5    b     20     5    5
6   b1     20     3    7
7   b2     20     2    8
8   b3     20     6    4
9   b4     20     2    8