如何直接替换列中的 NaN 值？

Question

我有一个像这样的 dataframe：

article_id          title
NaN                 title_1
NaN                 title_2
NaN                 title_3
'202102011404103'   title_4
'202102011404104'   title_5
NaN                 title_6

首先，我想添加一个条件“如果存在 NaN 值，请执行以下操作”（否则什么也不做）。 如果列中有 NaN 值：我想直接将增量值添加到列中的 NaN 值，如下所示：

x = 1
df.insert(0, 'article_id', range(x, x + len(df)))

但是我不知道如何将上面的代码直接添加到列article_id，只添加到NaN值。 我该怎么做？

预期 output：

article_id              title
    1                   title_1
    2                   title_2
    3                   title_3
    '202102011404103'   title_4
    '202102011404104'   title_5
    4                   title_6

Answer 1

您可以创建掩码以比较缺失值并通过sum与NaN计数的第一个值的范围：

m = df['article_id'].isna()

x = 1
df.loc[m, 'article_id'] =  range(x, x + m.sum())
print (df)
          article_id    title
0                  1  title_1
1                  2  title_2
2                  3  title_3
3  '202102011404103'  title_4
4  '202102011404104'  title_5
5                  4  title_6