在 Pandas 中将连续数值数据转换为离散数值数据

Question

我有一个 pandas 数据框 df，其中有一列包含连续的数值数据。

   A    
0   1.5   
1   15.0   
2   12.8   
3   23.2  
4   9.6

我想根据以下规则将连续变量替换为数值：

0-10=10
10-20=50
20-100=80

最终得到的dataframe应该是这样的：

    A    
0   10  
1   50   
2   50  
3   80  
4   10

我曾尝试使用pandas.cut(df, bins=[0,10,20,100], labels=[10,50,80])，但它返回一个分类列。我需要输出列是数字。

Answer 1

将to_numeric 添加到您的代码中

pd.to_numeric(pd.cut(df['A'], bins=[0,10,20,100], labels=[10,50,80]))
Out[54]: 
0    10
1    50
2    50
3    80
4    10
Name: A, dtype: int64