Python 我如何根据前一个连续数字改变一串数字

发布于04月08日

这是我的DataFrame:

import pandas as pd
df = pd.DataFrame(
    {
        'a': [1, 1, 1, 2, 2, 2, 2, 2, -1, -1, 2, 2, 2],
    }
)

预期输出:更改第a栏:

流程如下:

a)寻找1和2的条纹

b)如果连线2在连线1之后出现，则连线2变为1.

例如:

从第3行到第7行有一个2的条纹.在此之前的连胜是1的连胜.所以我想把行3到7改为1，这是df的预期输出.

这些是我的小伙伴:

# attempt 1
df['streak'] = df.a.ne(df.a.shift(1)).cumsum()
# attempt 2
df.loc[(df.a.eq(2)) & (df.a.shift(1).eq(1)), 'a'] = 1

推荐答案

您可以使用shift和ffill访问上一个组，然后使用boolean indexing:

# previous value
s = df['a'].shift()
# check if previous group is 1
m1 = s.where(df['a'].ne(s)).ffill().eq(1)
# is value 2?
m2 = df['a'].eq(2)

df.loc[m1&m2, 'a'] = 1

输出:

中间体:

    a    s  where  ffill     m1     m2
0   1  NaN    NaN    NaN  False  False
1   1  1.0    NaN    NaN  False  False
2   1  1.0    NaN    NaN  False  False
3   2  1.0    1.0    1.0   True   True
4   2  2.0    NaN    1.0   True   True
5   2  2.0    NaN    1.0   True   True
6   2  2.0    NaN    1.0   True   True
7   2  2.0    NaN    1.0   True   True
8  -1  2.0    2.0    2.0  False  False
9  -1 -1.0    NaN    2.0  False  False
10  2 -1.0   -1.0   -1.0  False   True
11  2  2.0    NaN   -1.0  False   True
12  2  2.0    NaN   -1.0  False   True