将带有分类值的pandas数据帧转换为二进制值

时间:2013-06-27 17:50:07

标签: python pandas scikit-learn

我正在尝试将分类数据转换为二进制数据,以便能够使用逻辑回归等算法进行分类。我想过使用'sklearn.preprocessing'模块中的OneHotEncoder,但问题是数据帧条目是A,B对具有不同长度的数组,每行都有一对不等于数组的相同长度数组其他行的长度。 OneHotEncoder 不接受像我这样的数据框

  

在[34]中:data.index

     

Out [34]:Index([train1,train2,train3,...,train7829,train7830,   train7831],dtype = object)

In [35]:  data.columns

Out[35]:  Index([A, B], dtype=object)

SampleID                      A                                B
train1      [2092.0, 1143.0, 390.0, ...]          [5651.0, 4449.0, 4012.0...]
train2      [3158.0, 3158.0, 3684.0, 3684.0....]  [2.0, 4.0, 2.0, 1.0...]
train3      [1699.0, 1808.0 ,...]                 [0.0, 1.0...]

所以,我想再次强调每个A和B对具有相同的长度,但长度在不同的对上是可变的。数据框包含数值,分类和二进制值。 我有另一个csv文件,其中包含有关每个条目类型的信息。我读过文件过滤掉两列中的分类条目,如下所示:

info=data_io.read_train_info()
col1=info.columns[0]
col2=info.columns[1]
info=info[(info[col1]=='Categorical')&(info[col2]=='Categorical')]

然后我使用info.index来过滤我的训练数据框

filtered = data.loc[info.index]

我编写了一个实用程序函数来更改每个数组的维度,以便我以后可以对它们进行编码

def setDim(df):
    for item in x[x.columns[0]].index:
        df[df.columns[0]][item].shape=(1,df[df.columns[0]][item].shape[0])
        df[df.columns[1]][item].shape=(1,df[df.columns[1]][item].shape[0])

setDim(filtered)

然后我想将每对数组合并成2行矩阵,这样我就可以将它传递给编码器,然后在编码后再将它们分开,如下所示:

import numpy as np
from sklearn.preprocessing import OneHotEncoder

def makeSparse(df):
   enc = OneHotEncoder()
   for i in df.index:
     cd=np.append(df['A'][i],df['B'][i],axis=0)
     a=enc.fit_transform(cd)
     df['A'][i] = a[0,:]
     df['B'][i] = a[1,:]

makeSparse(filtered)

在所有这些步骤之后获得稀疏的数据帧。我的问题是:

  1. 这是对这个数据帧进行编码的正确方法吗?(我非常怀疑)
  2. 如果没有,那么你提供什么替代品?
    非常感谢您抽出时间帮助我。

1 个答案:

答案 0 :(得分:1)

这是将数据转换为更好的repr来处理的好方法;用一些 整洁的应用技巧

In [72]: df
Out[72]: 
                               A                  B
train1         [2092, 1143, 390]  [5651, 449, 4012]
train2  [3158, 3158, 3684, 3684]       [2, 4, 2, 1]
train3              [1699, 1808]             [0, 1]

In [73]: concat(dict([ (x[0],x[1].apply(lambda y: Series(y))) for x in df.iterrows() ]))
Out[73]: 
             0     1     2     3
train1 A  2092  1143   390   NaN
       B  5651   449  4012   NaN
train2 A  3158  3158  3684  3684
       B     2     4     2     1
train3 A  1699  1808   NaN   NaN
       B     0     1   NaN   NaN