將 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)