首页 > 解决方案 > 使用 pandas 将 CSV 文件中的字符串转换为浮点数或整数?

问题描述

我正在尝试使用 tensorflow 和 keras 创建一个神经网络来预测加拿大车祸中的死亡人数。我从 CSV 文件导入了我的数据集,并将其中一列设置为标签。大多数列只包含浮点数,但其中一些列(例如性别)包含 M 或 F 等字符串。有没有办法可以将这些字符串转换为浮点数?

import tensorflow as tf
import numpy as np
import pandas as pd

df = pd.read_csv(path_to_csv)
for r in df.rows:
    print(type(r))
y_train = df["C_SEV"].copy()
x_train = df.drop(columns=["C_SEV"])


print(df["C_SEV"])

x_train = np.array(x_train)
x_train.flatten()

标签: pythonpython-3.xpandastensorflowkeras

解决方案


推荐阅读