机器学习-标签数值映射

Section I: Map Ordinary Categorical Label

第一部分：代码

import pandas as pd
import numpy as np

df=pd.DataFrame([['green','M',10.1,'class1'],
                 ['red','L',13.5,'class2'],
                 ['blue','XL',15.3,'class1']])

#Section 1: Encoding class labels via class-integer mapping dictionary
df.columns=['color','size','price','classlabel']
class_mapping={
   label:idx for idx,label in enumerate(np.unique(df['classlabel']))}

print("Transform Before:\n",df)
df['classlabel']=df['classlabel'].map(class_mapping)
print("After Transform:\n"<