python - 如何将字符串数据分类为整数？

Question

我需要对数据集特征的字符串值进行分类，以便我可以进一步将其用于其他用途，例如预测或绘图。

如何转换？

我找到了这个解决方案，但在这里我必须为该功能的每个唯一值手动键入代码。对于 2-3 个唯一值，没关系，但我有一个包含 50 多个国家/地区唯一值的功能，我无法为每个国家/地区编写代码。

def sex_class(x):
    if x == 'male':
        return 1
    else:
        return 0

这会将特征 - 性别中的男性值更改为 1，将女性值更改为 0。

Answer 1

您可以使用 scikit-learn LabelEncoder

#给定一个包含所有可能标签的列表
sex_classes = ['男性', '女性']
从sklearn导入预处理
le = 预处理.LabelEncoder()
le.fit(sex_classes)

这将为给定列表中的所有唯一值分配标签。您也可以将此标签编码器对象保存为 pickle 文件以供以后使用。