python - 有没有办法知道哪个分类值被赋予了什么标签？

Question

我正在处理包含“好”、“中”、“坏”的分类列的数据，现在我想知道哪个数字已分配给哪个类别，即中等分配为 1 还是 2？

Answer 1

如果您使用 sklearn ，请le.classes_给出枚举它们的类的名称：

from sklearn import preprocessing

mylist = ['r', 'z', 'd', 'd', 'r', 'a']

le = preprocessing.LabelEncoder()
le.fit_transform(mylist)

>>> array([2, 3, 1, 1, 2, 0])

le.classes_

>>> array(['a', 'd', 'r', 'z'], dtype='<U1')

您可以检查使用inverse_transform：

list(le.inverse_transform([i for i in range(len(le.classes_))]))

>>> ['a', 'd', 'r', 'z']