Python 在数据帧中填充2个特定唯一值之间的值

发布于11月30日

我有以下数据框:

index	Value
1	None
2	A
3	None
4	A
5	B
6	B
7	None
8	A
9	A
10	B

其思想是在A和B之间填零，以使该列中没有连续的A或B.

期望输出

index	Value
1	None
2	A
3	None
4	None
5	B
6	None
7	None
8	A
9	None
10	B

使用循环可以很容易地做到这一点，但由于我使用的是Pandas 和NumPy，我希望避免使用循环方法.

推荐答案

您可以 Select ffill向前传播非None，然后 Select shift，只保留与此新系列不同的值boolean indexing:

df.loc[df['Value'].eq(df['Value'].ffill().shift()), 'Value'] = None

或使用mask:

df['Value'] = df['Value'].mask(df['Value'].eq(df['Value'].ffill().shift()), None)

输出:

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

中间体:

   index Value ffill shift     eq
0      1  None  None  None   True
1      2     A     A  None  False
2      3  None     A     A  False
3      4     A     A     A   True
4      5     B     B     A  False
5      6     B     B     B   True
6      7  None     B     B  False
7      8     A     A     B  False
8      9     A     A     A   True
9     10     B     B     A  False