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

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)

效果如下：

注：Interval函数是数据区间的意思，0≤x≤10 就可以用Interval(0,10)

 upper_close是一个参数，默认是左闭右闭， upper_close=False就是右开