我有一个只有二进制值的熊猫系列
0
0 1
1 0
2 0
3 0
4 1
5 0
6 1
7 0
8 0
9 1
10 0
11 1
12 0
13 1
14 0
我想隐藏系列中的值,它是零,并且它周围有一个,即基本上改变1,0,1到1,1,1.
我需要的输出是:
0 1
1 0
2 0
3 0
4 1
5 1
6 1
7 0
8 0
9 1
10 1
11 1
12 1
13 1
14 0
我在这里尝试的是创建一个3的滚动窗口,并检查值是否是我需要的.有没有更好的方法来做这个?
>>> window = df.rolling(3, center=True)
>>> (df[0] | window.apply(lambda x: 1 if (x == [1,0,1]).all() else 0)[0].fillna(0)).astype(int)
注意:我也试过了换档功能.
最佳答案 使用
shift
表示布尔掩码,并按
numpy.where
设置1:
m1 = df[0].shift() == 1
m2 = df[0].shift(-1) == 1
m3 = df[0] == 0
df[0] = np.where(m1 & m2 & m3, 1, df[0])
print (df)
0
0 1
1 0
2 0
3 0
4 1
5 1
6 1
7 0
8 0
9 1
10 1
11 1
12 1
13 1
14 0