将数值数据更改为分类数据 - 熊猫

Question

我有一个pandas数据帧,其数字列为"amount".数量从0到20000不等.我想将其更改为定义范围的分类变量.因此,分类变量将是:

1. 介于0-1000美元之间
2. 1000-2000美元等等......直到19000-20000美元

我无法弄清楚如何更改列.我可以将其更改为二进制值,如下所示:

months["value"] = np.where(months['amount']>=450, 'yes', 'no') 

但是,如何对具有2个以上值的分类变量进行此操作？

Answer 1

你可以使用cut:

df = pd.DataFrame({'B':[4000,5000,4000,9000,5,11040]})

df['D'] = pd.cut(df['B'], range(0, 21000, 1000))
print (df)
       B               D
0   4000    (3000, 4000]
1   5000    (4000, 5000]
2   4000    (3000, 4000]
3   9000    (8000, 9000]
4      5       (0, 1000]
5  11040  (11000, 12000]