python - 将 float 转换为 int 并保留空值

Question

我有以下数据框，我想将“b”列中的值转换为整数

    a   b       c
0   1   NaN     3
1   5   7200.0  20
2   5   580.0   20

以下代码抛出异常“ValueError:无法将 NA 转换为整数”

df['b'] = df['b'].astype(int)

我如何仅将 float 转换为 int 并保留空值？

Answer 1

当您的系列包含 float 和 nan 并且您想要转换为整数时，当您尝试将 float 转换为 numpy 整数时会出现错误，因为存在 na 值。

不要做:

df['b'] = df['b'].astype(int)

从 pandas >= 0.24 开始，现在有一个内置的 pandas 整数。这确实允许整数 nan。注意 'Int64' 中的大写字母。这是 pandas 整数，而不是 numpy 整数。

那么，这样做:

df['b'] = df['b'].astype('Int64')

有关 pandas 整数 na 值的更多信息:
https://pandas.pydata.org/pandas-docs/stable/user_guide/gotchas.html#nan-integer-na-values-and-na-type-promotions