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

Question

我的 DataFrame 中有一个系列 (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 类型的，所以只要你有NaN ，Pandas 就会总是将你的列向下转换为 float 。 您可以使用Nullable Integer ，可从 Pandas 0.24.0 获得：

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 ，您可以用0屏蔽NaN ：

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