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

Question

我得到了要训练的 CSV 文件，但训练时出现错误。

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

Answer 1

显然您尝试将行“B”的值转换为浮点数。并且值“CBH”不能转换为浮点数。

Answer 2

Label Encoding可以将可用的值变成自己唯一的值。

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 Label Encoding Documentation