根据以前记录的匹配值创建新的 pandas 列

Question

我有一个带有列名称和 id 的 pandas 数据框，如下面的输入数据。我想创建一个新列“标签”，这样如果 id 比当前记录小一的记录具有相同的名称值，那么当前记录在标签列中的字母值将与前一个记录相同。如果它的名称有不同的值，那么我想递增到下一个字母值。我在下面提供了示例输出数据来说明我正在尝试做的事情。有谁知道用 pandas 或其他方式做到这一点的巧妙方法？

输入数据：

name id
cat   0
cat   1
dog   2
frog  3

输出数据：

label name id
A     cat   0
A     cat   1
B     dog   2
C     frog  3

Answer 1

这是一种方法：

from string import ascii_uppercase
from itertools import count



gen_letter = ((c * i for c in ascii_uppercase) for i in count(1))
r = []
for i, t in df.name.eq(df.name.shift()).items():
    if t:
        r.append(r[i - 1])
    else:
        r.append(next(gen_letter))

df['label'] = r

这个解决方案适用于许多标签，在您完成字母表后会变成：'AA'、'BB' ....