在pandas中使用反标记对对象进行解码

import pandas as pd

# 原始数据
data = ['apple', 'banana', 'orange', 'banana', 'apple', 'orange']
df = pd.DataFrame(data, columns=['fruits'])

# 对fruits列进行编码
df['fruits_encoded'] = df['fruits'].factorize()[0]

# 获取编码后的唯一值
labels = df['fruits_encoded'].unique()

# 反标记，将编码值转换为原始标签
df['fruits_decoded'] = pd.Series(labels).take(df['fruits_encoded'])

# 输出结果
print(df)

   fruits  fruits_encoded fruits_decoded
0   apple               0          apple
1  banana               1         banana
2  orange               2         orange
3  banana               1         banana
4   apple               0          apple
5  orange               2         orange