如何 map 在 pandas 的 id 列中的唯一项目

Question

label 用我自己的系列编码 pandas 中的一列

data = [[1,'A'],
        [1,'A'],
        [1,'B'],
        [2,'C']]

df = pd.DataFrame(data = data,columns = ['id','element'])
df['element'] = le.fit_transform(df['element'])

output

   id  element
0   1        0
1   2        0
2   3        1
3   4        2

这很好，但我也想要 p append 到字符串

   id  element
0   1       p0
1   2       p0
2   3       p1
3   4       p2

Answer 1

得到 output dataframe df后，您可以在末尾添加：

df.element = "p"+(df.element+1).map(str)

结果：

   id  element
0   1        p1
1   2        p1
2   3        p2
3   4        p3