如何根据另一列的值在dataframe中设置0和1

import pandas as pd

# 创建示例Dataframe
data = {'column1': [1, 2, 3, 4, 5],
        'column2': ['A', 'B', 'A', 'B', 'A']}
df = pd.DataFrame(data)

# 使用条件语句和逻辑运算符设置column1的值为0或1
df['column1'] = df['column2'].apply(lambda x: 1 if x == 'A' else 0)

# 打印更新后的Dataframe
print(df)

   column1 column2
0        1       A
1        0       B
2        1       A
3        0       B
4        1       A