连续数据的分类

Question

我有一个Pandas df，我用它在Scikit for Python中进行机器学习。其中一列是目标值，它是连续数据(从-10到+10不等)。

从目标列中，我想计算一个包含5个类的新列，其中每个类的行数相同，即如果我有1000行，我想将其分布到5个类中，每个类中大约有200行。

到目前为止，我已经在Excel中完成了这一点，与我的Python代码分开，但随着数据的增长，它变得越来越不实用。

在Excel中，我计算了百分位数，然后使用一些逻辑来构建类。

如何在Python中做到这一点?

Answer 1

#create data
import numpy as np
import pandas as pd
df = pd.DataFrame(20*np.random.rand(50, 1)-10, columns=['target'])   
#find quantiles
quantiles = df['target'].quantile([.2, .4, .6, .8])
#labeling of groups
df['group'] = 5
df['group'][df['target'] < quantiles[.8]] = 4
df['group'][df['target'] < quantiles[.6]] = 3       
df['group'][df['target'] < quantiles[.4]] = 2 
df['group'][df['target'] < quantiles[.2]] = 1 

Answer 2

寻找类似问题的答案，发现了这篇文章和下面的提示:熊猫之间有什么区别?Qcut和熊猫。

import numpy as np
import pandas as pd
#generate 1000 rows of uniform distribution between -10 and 10
rows = np.random.uniform(-10, 10, size = 1000)
#generate the discretization in 5 classes
rows_cut = pd.qcut(rows, 5)
classes = rows_cut.factorize()[0]