使用map函数将字符串类型变量转换为数字变量以便使用机器学习算法。
创建一个DataFrame
from pandas import DataFrame
df = DataFrame({'类别':['水果','水果','水果','蔬菜','蔬菜','肉类','肉类'],
'产地':['美国','中国','中国','中国','新西兰','新西兰','美国']})
df.head()
将类别转换为int
map_class = {'水果':1, '蔬菜':2, '肉类':3}
df['类别'] = df['类别'].map(map_class)
df