当等于特定值时移动列中的行

Question

当值等于 Column 中的特定值时，我想在 pandas df 中移动 rows。对于下面的 df，当 A == x.

中的值时，我试图将 Column B 中的值移动到 Column A

import pandas as pd

df = pd.DataFrame({
    'A' : [1,'x','x','x',5],
    'B' : ['x',2,3,4,'x'],
        })

这是我的尝试：

df = df.loc[df.A.shift(-1) == df.A.shift(1), 'x'] = df.A.shift(1)

预期输出：

Answer 1

您可以使用：

m = df.A.eq('x')
df[m]=df[m].shift(-1,axis=1)
print(df)

   A    B
0  1    x
1  2  NaN
2  3  NaN
3  4  NaN
4  5    x

Answer 2

您可以使用：

df[df.A=='x'] = df.shift(-1,axis=1)

print(df)

   A    B
0  1    x
1  2  NaN
2  3  NaN
3  4  NaN
4  5    x

Shift rows in a Column when equal to specific value