来源:旭玩手游网 更新:2024-02-19 16:00:59
用手机看
大家好,我是一名数据分析师,今天我将为大家评测对比pandas中新增一列并按条件赋值的不同方法。
1.使用.loc方法
首先,我们来看使用.loc方法新增一列并按条件赋值的方式。这种方法非常简单直观,适用于对数据框中的特定行进行操作。我们可以通过指定条件筛选出需要的行,并将新列的值赋给这些行。
以一个示例数据框为例,假设我们有一个销售数据表,包含了产品名称、销售额和销售数量。我们想要根据销售额来判断产品的销售等级,并新增一列"销售等级"。代码如下:
python import pandas as pd #创建示例数据框 data ={'产品名称':['A','B','C','D'], '销售额':[1000, 2000, 3000, 4000], '销售数量':[10, 20, 30, 40]} df = pd.DataFrame(data) #使用.loc方法新增一列并按条件赋值 df.loc[df['销售额']> 2000,'销售等级']='高' df.loc[df['销售额']<= 2000,'销售等级']='低' print(df)
运行结果如下:
产品名称 销售额销售数量销售等级 0 A 1000 10 低 1 B 2000 20 低 2 C 3000 30 高 3 D 4000 40 高
2.使用apply方法