pandas - 添加一个基于现有值的列(bin,qcut)

时间:2013-07-03 13:52:57

标签: python pandas binning

我正在慢慢地从 R 转移到 python + pandas ,我正面临一个我无法解决的问题......

我需要对一列中的值进行离散处理,方法是将它们分配到容器中,然后将具有这些容器名称的列添加到原始DataFrame。我正在尝试使用pandas.qcut,但生成的Categorical对象似乎与DataFrame的效果不佳。

一个例子:

import pandas as pd
df1 = pd.DataFrame(np.random.randn(10), columns=['a'])
df1['binned_a'] = pd.qcut(df1['a'],4)

现在尝试在describe上调用df1时,我看不到新列:

>>> df1.describe()
               a
count  10.000000
mean    0.594072
std     1.109981
min    -0.807307
25%    -0.304550
50%     0.545839
75%     1.189487
max     2.851922

然而,它显然是存在的:

>>> df1
          a          binned_a
0  0.190015   (-0.305, 0.546]
1  0.140227   (-0.305, 0.546]
2  1.380000    (1.189, 2.852]
3 -0.522530  [-0.807, -0.305]
4 -0.452810  [-0.807, -0.305]
5  2.851922    (1.189, 2.852]
6 -0.807307  [-0.807, -0.305]
7  0.901663    (0.546, 1.189]
8  1.010334    (0.546, 1.189]
9  1.249205    (1.189, 2.852]

我做错了什么?我想要的结果是获得一个包含4个唯一字符串值的列,用于描述垃圾箱(如R中的因素)。


修改

正如Dan正确发现的那样,summary()方法不会显示包含纯文本数据的列,因此解决了神秘的问题:)非常感谢!

1 个答案:

答案 0 :(得分:3)

我从未成为R用户,但如果我了解您,您希望将数据分组到箱中并描述每个箱。

In [9]: df.groupby('binned_a').describe().unstack()
Out[9]:               a                                                   \
                  count      mean       std       min       25%       50%   
binned_a                                                                    
(-0.113, 0.109]       2  0.025114  0.010264  0.017856  0.021485  0.025114   
(-0.337, -0.113]      2 -0.282838  0.056445 -0.322751 -0.302794 -0.282838   
(0.109, 0.563]        3  0.354481  0.214402  0.134978  0.250027  0.365076   
[-1.842, -0.337]      3 -1.003969  0.765167 -1.841622 -1.335073 -0.828523   


                   75%       max  
binned_a                              
(-0.113, 0.109]   0.028742  0.032371  
(-0.337, -0.113] -0.262882 -0.242925  
(0.109, 0.563]    0.464233  0.563390  
[-1.842, -0.337] -0.585142 -0.341762  

要完全避免使用分类,请参阅https://stackoverflow.com/a/17150734/1221924