假设我们有一个包含“姓名”、“年龄”和“性别”的数据框。我们要编写一个条件函数,该函数根据性别选择将“VIP”或“VVIP”添加到数据框中。
示例代码:
import pandas as pd
df = pd.DataFrame({'姓名': ['张三', '李四', '王五'],
'年龄': [25, 30, 35],
'性别': ['男', '女', '男']})
def add_vip_label(row):
if row['性别'] == '男':
return 'VIP'
elif row['性别'] == '女':
return 'VVIP'
else:
return '未知'
df['标签'] = df.apply(add_vip_label, axis=1) # 在数据框中添加“标签”列
print(df)
代码的输出结果将显示:
姓名 年龄 性别 标签
0 张三 25 男 VIP
1 李四 30 女 VVIP
2 王五 35 男 VIP
在“标签”列中,根据每个人的性别,我们已经按预期添加了它们的等级标签。