python - 无法将非有限值(NA 或 inf)转换为整数

Question

这个问题已经有答案了:

Convert Pandas column containing NaNs to dtype `int` (30 个回答)

已关闭 5 年前。

我有一个看起来像这样的数据框

   survived pclass  sex age sibsp   parch   fare    embarked
    0   1   1   female  29.0000 0   0   211.3375    S
    1   1   1   male    0.9167  1   2   151.5500    S
    2   0   1   female  2.0000  1   2   151.5500    S
    3   0   1   male    30.0000 1   2   151.5500    S
    4   0   1   female  25.0000 1   2   151.5500    S

我想将 'sex' 转换为 0, 1 编码，并使用 isnull 检查列中没有 NA

但是，在这一行我收到 ValueError: Cannot conversion non-finite value (NA or inf) to integer

df['sex'] = df['sex'].map({'female':0, 'male':1}).astype(int)

有什么建议吗？谢谢!

Answer 1

使用np.where

例如:

import numpy as np

df['sex'] = np.where(df['sex'] == 'female', 0, 1)