python – 如何根据条件更改pandas系列中的值

我有一个只有二进制值的熊猫系列

    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
点赞