pandas 用map将不同类别替换为整数

数据集:

 

 代码:

train=pd.read_csv('./1.csv')

train['y']=train['y'].map({'A':1,'B':2,'C':3,'D':4,'E':5})//将ABCDE字符进行替换。

print(train)

输出:

 

 map相当于替换将指定字符映射为指定数字，如果字符不在map内会被NaN替换。

修改后的代码:

train=pd.read_csv('./1.csv')

train['y']=train['y'].map({'A':1,'B':2,'C':3,'D':4,'E':5,'F':3})

print(train)

输出:

也可以映射为数字和字符。

train=pd.read_csv('./1.csv')

train['y']=train['y'].map({'A':1,'B':2,'C':3,'D':4,'E':5,'F':'w'})

print(train)