我正在尝试将分类数据转换为二进制数据,以便能够使用逻辑回归等算法进行分类。我想过使用'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)
在所有这些步骤之后获得稀疏的数据帧。我的问题是:
答案 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