如何在pandas中创建一个新列，该列是基于某个条件的另一列的总和？

import pandas as pd

# 创建示例DataFrame
data = {'column1': [1, 2, 3, 4, 5],
        'column2': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)

# 创建新列，基于条件的另一列的总和
condition = df['column1'] > 2  # 设置条件，例如column1大于2
total_sum = df.loc[condition, 'column2'].sum()  # 满足条件的column2值求和
df['new_column'] = total_sum  # 将总和赋值给新列

# 打印DataFrame
print(df)

   column1  column2  new_column
0        1       10         120
1        2       20         120
2        3       30         120
3        4       40         120
4        5       50         120