pandas--将字符串属性转换为int型

使用map函数将字符串类型变量转换为数字变量以便使用机器学习算法。

创建一个DataFrame

from pandas import DataFrame
df = DataFrame({'类别':['水果','水果','水果','蔬菜','蔬菜','肉类','肉类'],
                '产地':['美国','中国','中国','中国','新西兰','新西兰','美国']})
df.head()

将类别转换为int

map_class = {'水果':1, '蔬菜':2, '肉类':3}
df['类别'] = df['类别'].map(map_class)
df