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

Question

我的 DataFrame 中有一个系列 (month_addded)，如下所示:

9.0
12.0
12.0
nan
1.0

我希望所有的 float 都是整数，而 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)