来源:旭玩手游网 更新: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方法