python - 使用 LabelEncoder 转换数据

Question

我编写了这个函数来使用 LabelEncoder 转换分类特征

#convert columns to dummies with LabelEncoder
cols = ['ToolType', 'TestType', 'BatteryType']
#apply ene hot encoder
le = LabelEncoder()
for col in cols:
    data[col] = data[col].astype('|S') #convert object to str type before apply label encoder
    le.fit(ravel(data[col]))
    data[col] = le.transform(ravel(data[col]))

那些列中有空值，但有这样的错误

TypeError: ufunc 'isnan' not supported for the input types, and the inputs could not be safely coerced to any supported types according to the casting rule ''safe''

有谁知道如何帮助我解决这个问题？谢谢

Answer 1

此行正在转换为 numpy bytes_编码器不支持:

data[col] = data[col].astype('|S')

如果要转字符串，改 '|S'至 str :

data[col] = data[col].astype(str)

作为旁注，您可以使用 apply() 将循环减少到一行。和 fit_transform :

df[cols] = df[cols].astype(str).apply(le.fit_transform)