数据分桶：pandas实现case when的操作

最新推荐文章于 2025-03-23 18:44:35 发布

原创最新推荐文章于 2025-03-23 18:44:35 发布 · 2.2k 阅读

12 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #jupter

该博客介绍了如何使用`Interval`函数将数据分段，对商品价格进行0-150的四段划分，分别为0-10、10-50、50-100、100-150和150以上。通过`apply`方法将分段函数应用到DataFrame的`item_price`列，生成了`price_range`新列来展示价格范围。

def data_divide(x):
    if x in Interval(0,10,upper_closed=False):
        return "0-10"
    elif x in Interval(10,50,upper_closed=False):
        return "10-50"
    elif x in Interval(50,100,upper_closed=False):
        return "50-100"
    elif x in Interval(100,150,upper_closed=False):
        return "100-150"
    else:
        return "150+"
    
df["price_range"] = df["item_price"].apply(data_divide)

效果如下：