文章目录

• 一、问题描述
• 二、问题解决

一、问题描述

利用sklearn中的LabelEncoder函数进行编码后，想要得到原来的标签和现在编码之间的映射关系。

le = LabelEncoder()
labels_en = le.fit_transform(labels).astype(np.int64)

二、问题解决

可以新建一个字典，用循环更新：

res = {}
for cl in le.classes_:
    res.update({cl:le.transform([cl])[0]})

效果：