python - 在处理 csv 文件时将字符串转换为浮点值

Question

我得到了 CSV 文件进行训练，但训练错误即将到来。

ValueError：无法将字符串转换为浮点数：'CBH'

标签： pythonstringcsvtype-conversionvalueerror

Answer 1

标签编码可以将可用的值变成它们自己的唯一值。

import pandas as pd
import sklearn

df = pd.read_csv('file-path')

le = sklearn.preprocessing.LabelEncoding()
le.fit(df['code'])
df = df[le.transform(df['code'])]

# change back
df = df[le.inverse_transform(df['code'])]

Sklearn 标签编码文档