Python如何在依赖于另一列的数据帧内创建else if语句

import pandas as pd

# 创建示例数据帧
data = {'column1': [1, 2, 3, 4, 5],
        'column2': [5, 10, 15, 20, 25]}
df = pd.DataFrame(data)

# 使用条件语句创建新列column3
df['column3'] = ''
for index, row in df.iterrows():
    if row['column2'] > 10:
        df.at[index, 'column3'] = '大于10'
    elif row['column2'] == 10:
        df.at[index, 'column3'] = '等于10'
    else:
        df.at[index, 'column3'] = '小于10'

print(df)

   column1  column2 column3
0        1        5    小于10
1        2       10    等于10
2        3       15    大于10
3        4       20    大于10
4        5       25    大于10