有没有一种基于来自另一列的值来递增列的值的pythonic方法？

import pandas as pd

# 创建一个示例DataFrame
data = {
    'A': [1, 2, 3, 4],
    'B': [10, 20, 30, 40]
}
df = pd.DataFrame(data)

# 打印原始DataFrame
print("原始DataFrame:")
print(df)

# 假设我们想根据列'A'的值来递增列'B'的值
df['B'] = df['B'] + df['A']

# 打印修改后的DataFrame
print("\n修改后的DataFrame:")
print(df)

原始DataFrame:
   A   B
0  1  10
1  2  20
2  3  30
3  4  40

修改后的DataFrame:
   A   B
0  1  11
1  2  22
2  3  33
3  4  44

# 只有当列'A'的值大于2时才递增列'B'的值
df.loc[df['A'] > 2, 'B'] = df.loc[df['A'] > 2, 'B'] + df.loc[df['A'] > 2, 'A']

# 打印修改后的DataFrame
print("\n根据条件修改后的DataFrame:")
print(df)

根据条件修改后的DataFrame:
   A   B
0  1  10
1  2  20
2  3  36
3  4  48