df nan值的不同处理

时间:2020-08-18 10:15:21

标签: python pandas dataframe nan fillna

我有这样的df:

      A    B 
 0    1    5
 1    1    7
 2  NaN  NaN
 3    1    8
 4  NaN  NaN
 5  NaN  NaN
 6    2    6
 7    2    2
 8  NaN  NaN
 9  NaN  NaN
10    2    3

现在,我想用与外部不同的方式填充事件中的nan值。一个事件由A列标识,并且具有相同的值(在我的示例中为事件1和2)。 A列应仅包含事件中的事件编号。对于B列,应传送事件中的最后一个条目。 在事件之间,NaN值应设置为“ 0”。

我尝试了ffill()fillna(),但无法与我的条件相匹配。

预期结果:

      A    B 
 0    1    5
 1    1    7
 2    1    7
 3    1    8
 4    0    0
 5    0    0
 6    2    6
 7    2    2
 8    2    2
 9    2    2
10    2    3

感谢您的帮助:)

1 个答案:

答案 0 :(得分:0)

您可以将GroupBy.ffillSeries.mask创建的助手组一起使用,并通过反冲值进行比较以防止组之间遗漏值,最后将丢失的值替换为0和整数:

s = df.A.ffill()
g = df.A.mask(s.eq(df.A.bfill()), s)
df = df.groupby(g).ffill().fillna(0).astype(int)
print (df)
    A  B
0   1  5
1   1  7
2   1  7
3   1  8
4   0  0
5   0  0
6   2  6
7   2  2
8   2  2
9   2  2
10  2  3