python - 如果它们之间的差异小于 pandas 中的固定值，则将行值替换为先前的值

Question

我有一个类似的数据框

Index  Value
0        5
1        Nan
2        6
3        16
4        23
5        32
6        33
7        34

如果与上一行的差异小于 2，则当前行值必须仅等于上一行。

所以我想要这样

Index  Value
0        5
1        Nan
2        6
3        16
4        23
5        32
6        32
7        32

I have tried : 

test = np.where(df.Value.notnull() & df.Value.shift(-1).notnull() & ((df.Value - df.Value.shift(-1)) < 2),df.Value.shift(-1),df.Value)

它没有提供所需的输出，我哪里出错了？

编辑:

您好，我的要求有一个小修改。如果前一个值为空，我希望将它与之前的非空值进行比较并计算差值。所以在这种情况下，我希望它是

Index  Value
0        5
1        Nan
2        5
4        23
5        32
6        32
7        32

Index-2 的值应该是 5 而不是 6

Answer 1

一种方法是使用diff 和cumsum 创建组键，然后我们使用transform

s=df.Value.diff()
df.groupby((s.isnull()|s.gt(2)).cumsum()).Value.transform('first')
0     5.0
1     NaN
2     6.0
3    16.0
4    23.0
5    32.0
6    32.0
7    32.0
Name: Value, dtype: float64

---

更新呵呵

s=df.Value.diff()
print(df.groupby((s.gt(2)).cumsum()).Value.transform('first').mask(df.Value.isnull()))
0     5.0
1     NaN
2     5.0
3    16.0
4    23.0
5    32.0
6    32.0
7    32.0
Name: Value, dtype: float64