将列添加到数据框中，以执行逐个元素的操作

import pandas as pd

# 创建一个示例数据框
data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [20, 22, 21]
}
df = pd.DataFrame(data)

# 直接赋值添加新列
df['AgePlusOne'] = df['Age'] + 1

# 使用assign()方法添加新列
df = df.assign(AgePlusTwo=df['Age'] + 2)

# 使用计算结果赋值添加新列
df['IsAdult'] = df['Age'] >= 18

print(df)

      Name  Age  AgePlusOne  AgePlusTwo  IsAdult
0    Alice   20          21          22     True
1      Bob   22          23          24     True
2  Charlie   21          22          23     True

df = df.drop(columns=['AgePlusOne'])

df['Age'] = df['Age'] + 1

if 'AgePlusTwo' in df.columns:
    print("列存在")
else:
    print("列不存在")