编写一个条件函数，将值添加到数据框中。 _编程开发

编写一个条件函数，将值添加到数据框中。

创始人

2024-12-07 10:01:05

0次

假设我们有一个包含“姓名”、“年龄”和“性别”的数据框。我们要编写一个条件函数，该函数根据性别选择将“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

在“标签”列中，根据每个人的性别，我们已经按预期添加了它们的等级标签。

上一篇：编写一个条件递归SQL查询来查找在一个零售商处不可用的商品，但可以在其他零售商处提供的商品。

下一篇：编写一个条件循环来求和列表的一个子集，并返回二进制值

编写一个条件函数，将值添加到数据框中。

相关内容

热门资讯