在csv文件上工作时将字符串转换为浮点值

问题描述

我有要训练的CSV文件，但是当训练错误来临时。

ValueError：无法将字符串转换为float：'CBH'

解决方法

标签编码可以将可用值转换为自己的唯一值。

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

,

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