将 pandas 系列转换为具有 NaN 值的 int

Question

我的数据框中有一个系列（month_addded），如下所示：

9.0
12.0
12.0
nan
1.0

我希望所有浮点数都是整数，而 NaN 保持原样。我这样做了：

for i in df['month_added']:
    if i > 0:
        i=int(i)

但它什么也没做。

Answer 1

NaN是 float 类型，所以只要你有 ，Pandas 总是会向下转换你的列以浮动NaN。您可以使用Pandas 0.24.0 中提供的Nullable Integer ：

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

如果不可能，您可以强制Object输入（不推荐）：

df['month_added'] = pd.Series([int(x) if x > 0 else x for x in df.month_added], dtype='O')

或者由于您的数据为正且NaN，您可以NaN使用进行掩码0：

df['month_added'] = df['month_added'].fillna(0).astype(int)