根据groupby shift更改列的值，并创建新的列- Pandas

import pandas as pd

# 创建一个示例DataFrame
data = {'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
        'B': [1, 2, 3, 4, 5, 6, 7, 8]}
df = pd.DataFrame(data)

# 根据列A分组，并对列B进行位移操作
df['Shifted_B'] = df.groupby('A')['B'].shift(1)

# 创建新的列，并根据条件修改列的值
def change_value(row):
    if row['A'] == 'foo':
        return row['B'] + 10
    else:
        return row['B'] - 10

df['New_Column'] = df.apply(change_value, axis=1)

# 打印结果
print(df)

     A  B  Shifted_B  New_Column
0  foo  1        NaN          11
1  bar  2        NaN          -8
2  foo  3        1.0          13
3  bar  4        2.0          -6
4  foo  5        3.0          15
5  bar  6        4.0          -4
6  foo  7        5.0          17
7  foo  8        7.0          18