对符合给定条件的值求和，并创建新的数据框(python)

import pandas as pd

data = {'A': [1, 2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50],
        'C': [100, 200, 300, 400, 500]}
df = pd.DataFrame(data)

condition = df['A'] > 2  # 以列A大于2为例
sum_value = df.loc[condition, 'B'].sum()  # 求和列B中符合条件的值

new_df = df.loc[condition].copy()  # 复制符合条件的行
new_df['Sum'] = sum_value  # 添加一列来存储求和结果